[Fix] Fix collect_env without opencv

fanqiNO1 · Nov 22, 2023 · 2c5183f · 2c5183f
1 parent bdd653a
commit 2c5183f
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/mmengine/utils/dl_utils/collect_env.py b/mmengine/utils/dl_utils/collect_env.py
@@ -5,7 +5,6 @@
 import sys
 from collections import OrderedDict, defaultdict
 
-import cv2
 import numpy as np
 import torch
 
@@ -136,7 +135,12 @@ def collect_env():
     except ModuleNotFoundError:
         pass
 
-    env_info['OpenCV'] = cv2.__version__
+    try:
+        import cv2
+        env_info['OpenCV'] = cv2.__version__
+    except ModuleNotFoundError:
+        pass
+
     env_info['MMEngine'] = mmengine.__version__
 
     return env_info