Update cv2.imread() patch with flags argument (#7287)

ultralytics · Apr 5, 2022 · 2181ef3 · 2181ef3
1 parent 8d0291f
commit 2181ef3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/utils/general.py b/utils/general.py
@@ -925,8 +925,8 @@ def increment_path(path, exist_ok=False, sep='', mkdir=False):
 imshow_ = cv2.imshow  # copy to avoid recursion errors
 
 
-def imread(path):
-    return cv2.imdecode(np.fromfile(path, np.uint8), cv2.IMREAD_COLOR)
+def imread(path, flags=cv2.IMREAD_COLOR):
+    return cv2.imdecode(np.fromfile(path, np.uint8), flags)
 
 
 def imwrite(path, im):