Unverified Commit cd634a8f authored by wfng92's avatar wfng92 Committed by GitHub
Browse files

Check for all different packages of opencv (#2901)


Co-authored-by: default avatarPatrick von Platen <patrick.v.platen@gmail.com>
parent 7447f75b
......@@ -175,8 +175,21 @@ if _onnx_available:
# (sayakpaul): importlib.util.find_spec("opencv-python") returns None even when it's installed.
# _opencv_available = importlib.util.find_spec("opencv-python") is not None
try:
_opencv_version = importlib_metadata.version("opencv-python")
_opencv_available = True
candidates = (
"opencv-python",
"opencv-contrib-python",
"opencv-python-headless",
"opencv-contrib-python-headless",
)
_opencv_version = None
for pkg in candidates:
try:
_opencv_version = importlib_metadata.version(pkg)
break
except importlib_metadata.PackageNotFoundError:
pass
_opencv_available = _opencv_version is not None
if _opencv_available:
logger.debug(f"Successfully imported cv2 version {_opencv_version}")
except importlib_metadata.PackageNotFoundError:
_opencv_available = False
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment