From fd802910ecdb96180c2fa9b0783c75d4dc49ce53 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 3 Aug 2019 20:30:25 +0200 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 cefd9432..e559f987 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -285,7 +285,7 @@ class LoadImagesAndLabels(Dataset): # for training/testing h, w, _ = img.shape img = cv2.resize(img, (int(w * r), int(h * r)), interpolation=cv2.INTER_LINEAR) # INTER_LINEAR fastest - if self.n < 3000: # cache into memory if image count < 3000 + if self.n < 5000: # cache into memory if image count < 5000 self.imgs[index] = img # Augment colorspace