* Update datasets.py * comment Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>modifyDataloader
@@ -181,7 +181,7 @@ class LoadImages: # for inference | |||
else: | |||
# Read image | |||
self.count += 1 | |||
img0 = cv2.imread(path) # BGR | |||
img0 = cv2.imread(path, -1) # BGR (-1 is IMREAD_UNCHANGED) | |||
assert img0 is not None, 'Image Not Found ' + path | |||
print(f'image {self.count}/{self.nf} {path}: ', end='') | |||