From 35492722441f25dc9af0526b7849e322a048923f Mon Sep 17 00:00:00 2001 From: iperov Date: Fri, 15 Mar 2019 11:03:37 +0400 Subject: [PATCH] fix for final --- mainscripts/Sorter.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/mainscripts/Sorter.py b/mainscripts/Sorter.py index 2b4e0d84a..8f5e82b01 100644 --- a/mainscripts/Sorter.py +++ b/mainscripts/Sorter.py @@ -635,6 +635,15 @@ def sort_final(input_path, include_by_blur=True): 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] + if include_by_blur: for g in io.progress_bar_generator ( range (grads), "Sort by blur"): img_list = yaws_sample_list[g] @@ -642,10 +651,6 @@ def sort_final(input_path, include_by_blur=True): continue 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: trash_img_list += img_list[sharpned_imgs_per_grad:]