cv2.imread(img, -1) for IMREAD_UNCHANGED (#3379)

* Update datasets.py

* comment

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
tudouleiandGlenn Jocher authored and GitHub committed 2021-05-29 21:12:01 +02:00
1 parent 57f773b0ae
commit 21a9607e00
1 file changed
+1 -1
+1 -1
View File
@@ -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='')