updates
This commit is contained in:
parent
b2abd46437
commit
401a615d34
|
@ -199,7 +199,7 @@ class LoadImagesAndLabels(Dataset): # for training/testing
|
||||||
except:
|
except:
|
||||||
print('Warning: missing labels for %s' % self.img_files[i])
|
print('Warning: missing labels for %s' % self.img_files[i])
|
||||||
pass # missing label file
|
pass # missing label file
|
||||||
assert len(np.concatenate(self.labels, 0)) > 0, 'No labels found. Check label paths.'
|
assert len(np.concatenate(self.labels, 0)) > 0, 'No labels found. Incorrect label paths provided.'
|
||||||
|
|
||||||
def __len__(self):
|
def __len__(self):
|
||||||
return len(self.img_files)
|
return len(self.img_files)
|
||||||
|
|
Loading…
Reference in New Issue