github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-08 05:51:40 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

sorter: fix final sort

Browse source
This commit is contained in:
iperov 2019-03-16 09:20:55 +04:00
parent 5076feb90f
commit 7c66503382
1 changed files with 3 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

17
mainscripts/Sorter.py
View file

@ -632,21 +632,10 @@ def sort_final(input_path, include_by_blur=True):
lack = imgs_per_grad - img_list_len
total_lack += max(lack, 0)
imgs_per_grad += total_lack // grads
sharpned_imgs_per_grad = imgs_per_grad*10
for g in io.progress_bar_generator ( range (grads), "Normalizing"):
img_list = yaws_sample_list[g]
if img_list is None:
continue
if len(img_list) > imgs_per_grad*2:
trash_img_list += img_list[len(img_list) // 2:]
img_list = img_list[0: len(img_list) // 2]
yaws_sample_list[g] = img_list
imgs_per_grad += total_lack // grads
if include_by_blur:
sharpned_imgs_per_grad = imgs_per_grad*10
for g in io.progress_bar_generator ( range (grads), "Sort by blur"):
img_list = yaws_sample_list[g]
if img_list is None:
@ -659,7 +648,7 @@ def sort_final(input_path, include_by_blur=True):
img_list = img_list[0:sharpned_imgs_per_grad]
yaws_sample_list[g] = img_list
for g in io.progress_bar_generator ( range (grads), "Sort by hist"):
img_list = yaws_sample_list[g]
if img_list is None: