From 543e9e560c9e44e358d922c75e28abb398ed5cd4 Mon Sep 17 00:00:00 2001 From: pierrezeb <36326642+pierrezeb@users.noreply.github.com> Date: Thu, 7 Feb 2019 16:43:47 +0900 Subject: [PATCH] fix split_ids (returned only left square!!) Former-commit-id: a357099e4eace375489f18babb2947d9103733fd --- utils/load.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/load.py b/utils/load.py index 8317ffc..e5670e0 100644 --- a/utils/load.py +++ b/utils/load.py @@ -17,7 +17,7 @@ def get_ids(dir): def split_ids(ids, n=2): """Split each id in n, creating n tuples (id, k) for each id""" - return ((id, i) for i in range(n) for id in ids) + return ((id, i) for id in ids for i in range(n)) def to_cropped_imgs(ids, dir, suffix, scale):