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:]