Use the Dummy opencv to return an error if opencv not installed

tryolabs · Feb 26, 2024 · ed273fd · ed273fd
1 parent bed08ed
commit ed273fd
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/norfair/drawing/fixed_camera.py b/norfair/drawing/fixed_camera.py
@@ -1,6 +1,11 @@
 from typing import Union
 
-import cv2
+try:
+    import cv2
+except ImportError:
+    from norfair.utils import DummyOpenCVImport
+
+    cv2 = DummyOpenCVImport()
 import numpy as np
 
 from norfair.camera_motion import HomographyTransformation, TranslationTransformation

diff --git a/norfair/drawing/path.py b/norfair/drawing/path.py
@@ -1,7 +1,12 @@
 from collections import defaultdict
 from typing import Callable, Optional, Sequence, Tuple
 
-import cv2
+try:
+    import cv2
+except ImportError:
+    from norfair.utils import DummyOpenCVImport
+
+    cv2 = DummyOpenCVImport()
 import numpy as np
 
 from norfair.camera_motion import HomographyTransformation, TranslationTransformation