updates
This commit is contained in:
parent
b2abd46437
commit
401a615d34
|
@ -199,7 +199,7 @@ class LoadImagesAndLabels(Dataset): # for training/testing
|
|||
except:
|
||||
print('Warning: missing labels for %s' % self.img_files[i])
|
||||
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):
|
||||
return len(self.img_files)
|
||||
|
|
Loading…
Reference in New Issue