testing with a opencv-python that is smaller in size

erdogant · Jan 8, 2025 · a0133e5 · a0133e5
1 parent 82309c2
commit a0133e5
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/clustimage/clustimage.py b/clustimage/clustimage.py
@@ -25,7 +25,6 @@
 import os
 import logging
 import fnmatch
-import cv2
 import matplotlib.pyplot as plt
 from matplotlib import offsetbox
 
@@ -50,6 +49,14 @@
 # Register HEIF opener for Pillow
 register_heif_opener()
 
+try:
+    import cv2
+except ImportError:
+    raise ImportError(
+        "The 'opencv-python' library is not installed. Please install it using the following command:\n"
+        "pip install opencv-python")
+
+
 # Configure the logger
 logger = logging.getLogger('')
 [logger.removeHandler(handler) for handler in logger.handlers[:]]

diff --git a/requirements.txt b/requirements.txt
@@ -11,9 +11,9 @@ scikit-learn
 matplotlib
 scikit-image
 scipy
-opencv-python
 pypickle
 imagehash
 pypickle
 ismember
 pillow-heif
+opencv-python-headless
diff --git a/requirements_additional.txt b/requirements_additional.txt
@@ -1,3 +1,4 @@
 piexif
 geopy
 folium
+opencv-python