david/DeepFaceLab_Linux
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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
mainscripts/Sorter.py
+9 -4
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:]