github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-10 07:13:39 -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

13
mainscripts/Sorter.py
View file

@ -633,20 +633,9 @@ def sort_final(input_path, include_by_blur=True):
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
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: