From 39d247d7e85661c17a22083494d3be6b2308daf0 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 25 Oct 2019 10:55:08 -0500 Subject: [PATCH] updates --- utils/datasets.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/datasets.py b/utils/datasets.py index 7f2c9c04..5382d165 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -511,7 +511,7 @@ def load_image(self, index): img = cv2.imread(img_path) # BGR assert img is not None, 'Image Not Found ' + img_path r = self.img_size / max(img.shape) # size ratio - if self.augment and r < 1.0: # if training (NOT testing), downsize to inference shape + if self.augment: # if training (NOT testing), downsize to inference shape h, w, _ = img.shape img = cv2.resize(img, (int(w * r), int(h * r)), interpolation=cv2.INTER_LINEAR) # _LINEAR fastest return img