diff --git a/utils/datasets.py b/utils/datasets.py index 13ab07fc..8587c539 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -277,7 +277,11 @@ class LoadImagesAndLabels(Dataset): # for training/testing if img is None: img = cv2.imread(img_path) # BGR 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 # Augment colorspace