From ff3a208e801389d776693a7412068f4835ecf98b Mon Sep 17 00:00:00 2001 From: iperov Date: Mon, 7 Jan 2019 14:57:46 +0400 Subject: [PATCH] fix sort by final --- mainscripts/Sorter.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/mainscripts/Sorter.py b/mainscripts/Sorter.py index 26d0e79..cef5858 100644 --- a/mainscripts/Sorter.py +++ b/mainscripts/Sorter.py @@ -553,10 +553,12 @@ def sort_final(input_path): imgs_per_grad = 15 sharpned_imgs_per_grad = imgs_per_grad*10 + grads_space = np.linspace (-255,255,grads) + yaws_sample_list = [None]*grads - for g in tqdm ( range (grads), desc="Sort by yaw" ): - yaw = -grads+1 + g*2 - next_yaw = -grads+1 + (g+1)*2 + for g in tqdm ( range(grads), desc="Sort by yaw" ): + yaw = grads_space[g] + next_yaw = grads_space[g+1] if g < grads-1 else yaw yaw_samples = [] for img in img_list: