updates
This commit is contained in:
parent
001193b9c7
commit
8f9ab337b0
|
@ -227,9 +227,9 @@ class LoadImagesAndLabels(Dataset): # for training/testing
|
||||||
img = self.imgs[index]
|
img = self.imgs[index]
|
||||||
if img is None:
|
if img is None:
|
||||||
img = cv2.imread(img_path) # BGR
|
img = cv2.imread(img_path) # BGR
|
||||||
|
assert img is not None, 'File Not Found ' + img_path
|
||||||
if self.n < 1001:
|
if self.n < 1001:
|
||||||
self.imgs[index] = img # cache image into memory
|
self.imgs[index] = img # cache image into memory
|
||||||
assert img is not None, 'File Not Found ' + img_path
|
|
||||||
|
|
||||||
# Augment colorspace
|
# Augment colorspace
|
||||||
augment_hsv = True
|
augment_hsv = True
|
||||||
|
|
Loading…
Reference in New Issue