From d1abe51876fafff023cae5bc780af8ee0985a363 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 17 Jul 2019 14:19:09 +0200 Subject: [PATCH] updates --- utils/datasets.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/utils/datasets.py b/utils/datasets.py index 405dd710..6f475a49 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -278,11 +278,11 @@ class LoadImagesAndLabels(Dataset): # for training/testing img = cv2.imread(img_path) # BGR assert img is not None, 'File Not Found ' + img_path r = self.img_size / max(img.shape) # size ratio - if r < 1: # downsize is target shape is smaller + if r < 1: # downsize if target shape is smaller h, w, _ = img.shape img = cv2.resize(img, (int(w * r), int(h * r)), interpolation=cv2.INTER_AREA) - if self.n < 3000: # cache images into memory if image count < 3000 + if self.n < 3000: # cache into memory if image count < 3000 self.imgs[index] = img # Augment colorspace