mirrors/mmengine
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmengine.git synced 2025-06-03 21:54:44 +08:00
Code Issues Projects Releases Wiki Activity

[Fix] Fix collect_env without opencv (#1434)

Browse Source
This commit is contained in:
fanqiNO1 2023-11-22 10:50:23 +08:00 committed by GitHub
parent bdd653a8c3
commit 246ec1ff35
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mmengine/utils/dl_utils/collect_env.py
View File

@ -5,7 +5,6 @@ import subprocess
import sys import sys
from collections import OrderedDict, defaultdict from collections import OrderedDict, defaultdict
import cv2
import numpy as np import numpy as np
import torch import torch
@ -136,7 +135,12 @@ def collect_env():
except ModuleNotFoundError: except ModuleNotFoundError:
pass pass
env_info['OpenCV'] = cv2.__version__ try:
import cv2
env_info['OpenCV'] = cv2.__version__
except ModuleNotFoundError:
pass
env_info['MMEngine'] = mmengine.__version__ env_info['MMEngine'] = mmengine.__version__
return env_info return env_info