This commit is contained in:
Glenn Jocher 2019-05-21 12:59:04 +02:00
parent b2abd46437
commit 401a615d34
1 changed files with 1 additions and 1 deletions

View File

@ -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)