Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-05-21 14:01:06 +02:00
parent 401a615d34
commit 0effcd02bf
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
utils/datasets.py
View File

@ -194,11 +194,14 @@ class LoadImagesAndLabels(Dataset): # for training/testing
for i, file in enumerate(iter): for i, file in enumerate(iter):
try: try:
with open(file, 'r') as f: with open(file, 'r') as f:
self.labels[i] = np.array([x.split() for x in f.read().splitlines()], dtype=np.float32) l = np.array([x.split() for x in f.read().splitlines()], dtype=np.float32)
assert self.labels[i].shape[1] == 5, 'corrupted labels file: %s' % file if l.shape[0]:
assert l.shape[1] == 5, '> 5 label columns: %s' % file
assert (l >= 0).all(), 'negative labels: %s' % file
assert (l[:, 1:] <= 1).all(), 'non-normalized or out of bounds coordinate labels: %s' % file
self.labels[i] = l
except: except:
print('Warning: missing labels for %s' % self.img_files[i]) print('Warning: missing labels for %s' % self.img_files[i]) # missing label file
pass # missing label file
assert len(np.concatenate(self.labels, 0)) > 0, 'No labels found. Incorrect label paths provided.' assert len(np.concatenate(self.labels, 0)) > 0, 'No labels found. Incorrect label paths provided.'
def __len__(self): def __len__(self):