This commit is contained in:
Glenn Jocher 2019-11-19 20:42:44 -08:00
parent 253e746d30
commit 429d44282c
1 changed files with 1 additions and 1 deletions

View File

@ -367,7 +367,7 @@ class LoadImagesAndLabels(Dataset): # for training/testing
b[[1, 3]] = np.clip(b[[1, 3]], 0, h) b[[1, 3]] = np.clip(b[[1, 3]], 0, h)
assert cv2.imwrite(f, img[b[1]:b[3], b[0]:b[2]]), 'Failure extracting classifier boxes' assert cv2.imwrite(f, img[b[1]:b[3], b[0]:b[2]]), 'Failure extracting classifier boxes'
else: else:
ne += 1 # file empty ne += 1 # print('empty labels for image %s' % self.img_files[i]) # file empty
pbar.desc = 'Reading labels (%g found, %g missing, %g empty for %g images)' % (nf, nm, ne, n) pbar.desc = 'Reading labels (%g found, %g missing, %g empty for %g images)' % (nf, nm, ne, n)
assert nf > 0, 'No labels found. Recommend correcting image and label paths.' assert nf > 0, 'No labels found. Recommend correcting image and label paths.'