github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-06 13:02:15 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix for final

Browse source
This commit is contained in:
iperov 2019-03-15 11:03:37 +04:00
parent 1beb2f07f0
commit 3549272244
1 changed files with 9 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

13
mainscripts/Sorter.py
View file

@ -635,6 +635,15 @@ def sort_final(input_path, include_by_blur=True):
imgs_per_grad += total_lack // grads imgs_per_grad += total_lack // grads
sharpned_imgs_per_grad = imgs_per_grad*10 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]
if include_by_blur: if include_by_blur:
for g in io.progress_bar_generator ( range (grads), "Sort by blur"): for g in io.progress_bar_generator ( range (grads), "Sort by blur"):
img_list = yaws_sample_list[g] img_list = yaws_sample_list[g]
@ -642,10 +651,6 @@ def sort_final(input_path, include_by_blur=True):
continue continue
img_list = sorted(img_list, key=operator.itemgetter(1), reverse=True) img_list = sorted(img_list, key=operator.itemgetter(1), reverse=True)
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]
if len(img_list) > sharpned_imgs_per_grad: if len(img_list) > sharpned_imgs_per_grad:
trash_img_list += img_list[sharpned_imgs_per_grad:] trash_img_list += img_list[sharpned_imgs_per_grad:]