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

updates

This commit is contained in:
Glenn Jocher 2019-07-17 14:00:50 +02:00
parent 9d54a268c9
commit 34ddceea89
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
utils/datasets.py
View File

@ -277,7 +277,11 @@ class LoadImagesAndLabels(Dataset): # for training/testing
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 assert img is not None, 'File Not Found ' + img_path
if self.n < 1001: if self.n < 2001:
r = self.img_size / max(img.shape) # ratio
if r < 1: # downsize image and cache in RAM for future
h, w, _ = img.shape
img = cv2.resize(img, (int(w*r), int(h*r)), interpolation=cv2.INTER_AREA) # resized, no border
self.imgs[index] = img # cache image into memory self.imgs[index] = img # cache image into memory
# Augment colorspace # Augment colorspace