Laurent/REVA-QCAV
1
0
Fork 0
mirror of https://github.com/Laurent2916/REVA-QCAV.git synced 2024-09-19 19:45:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed major data loading bug : #46 from pierrezeb/patch-1

Browse source 
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:
milesial 2019-02-07 13:30:53 +01:00 committed by GitHub
parent 0f5e7cb0f9 543e9e560c
commit 74f825ce06
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
utils/load.py
View file

@ -17,7 +17,7 @@ def get_ids(dir):
def split_ids(ids, n=2): def split_ids(ids, n=2):
"""Split each id in n, creating n tuples (id, k) for each id""" """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): def to_cropped_imgs(ids, dir, suffix, scale):