From 8f9ab337b070b80dae3e51ff8c6948771087e119 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 23 May 2019 13:19:49 +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 49194207..991118a8 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -227,9 +227,9 @@ class LoadImagesAndLabels(Dataset): # for training/testing img = self.imgs[index] if img is None: img = cv2.imread(img_path) # BGR + assert img is not None, 'File Not Found ' + img_path if self.n < 1001: self.imgs[index] = img # cache image into memory - assert img is not None, 'File Not Found ' + img_path # Augment colorspace augment_hsv = True