fix split_ids (returned only left square!!)

Former-commit-id: a357099e4eace375489f18babb2947d9103733fd
2019-02-07 16:43:47 +09:00 · 2019-02-07 16:43:47 +09:00 · 543e9e560c
parent 0f5e7cb0f9
commit 543e9e560c
1 changed files with 1 additions and 1 deletions
--- 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):