mirror of
https://github.com/Laurent2916/REVA-QCAV.git
synced 2024-11-09 15:02:03 +00:00
Fixed major data loading bug : #46 from pierrezeb/patch-1
Fix split_ids (returned only (id, 0)) The network is now training on both squares and not only the left part of the image Former-commit-id: e2e46ce509382a45b1db4e1f639aeed568f6cb3e
This commit is contained in:
commit
74f825ce06
|
@ -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):
|
||||
|
|
Loading…
Reference in a new issue