From 01be8ffae04be2f0dc7b16dcddbf1acadc7a7d27 Mon Sep 17 00:00:00 2001 From: naelsondouglas Date: Fri, 15 Oct 2021 02:53:27 -0300 Subject: [PATCH] removed a case where the code was relying into the manual indexer handler pitfall --- evaluate.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/evaluate.py b/evaluate.py index 1d5abe7a..c6273914 100644 --- a/evaluate.py +++ b/evaluate.py @@ -135,8 +135,7 @@ def ffwd_different_dimensions(in_path, out_path, checkpoint_dir, device_t=DEVICE, batch_size=4): in_path_of_shape = defaultdict(list) out_path_of_shape = defaultdict(list) - for i in range(len(in_path)): - in_image = in_path[i] + for i, in_image in enumerate(in_path): out_image = out_path[i] shape = "%dx%dx%d" % get_img(in_image).shape in_path_of_shape[shape].append(in_image)