Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-03-05 16:13:40 +01:00
parent e5dc942fee
commit 3e6800fbc9
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
utils/datasets.py
View File

@ -37,7 +37,7 @@ class LoadImages: # for inference
# Read image # Read image
img0 = cv2.imread(img_path) # BGR img0 = cv2.imread(img_path) # BGR
assert img0 is not None, 'Failed to load ' + img_path assert img0 is not None, 'File Not Found ' + img_path
# Padded resize # Padded resize
img, _, _, _ = letterbox(img0, height=self.height) img, _, _, _ = letterbox(img0, height=self.height)
@ -106,7 +106,7 @@ class LoadImagesAndLabels: # for training
self.multi_scale = multi_scale self.multi_scale = multi_scale
self.augment = augment self.augment = augment
assert self.nB > 0, 'No images found in %s' % path assert self.nF > 0, 'No images found in %s' % path
def __iter__(self): def __iter__(self):
self.count = -1 self.count = -1
@ -134,8 +134,7 @@ class LoadImagesAndLabels: # for training
label_path = self.label_files[self.shuffled_vector[files_index]] label_path = self.label_files[self.shuffled_vector[files_index]]
img = cv2.imread(img_path) # BGR img = cv2.imread(img_path) # BGR
if img is None: assert img is not None, 'File Not Found ' + img_path
continue
augment_hsv = True augment_hsv = True
if self.augment and augment_hsv: if self.augment and augment_hsv: