fixes opencv issue with avif/heic

Author: joaomarcoscrs

requirements.txt

@@ -6,7 +6,6 @@ matplotlib
 numpy>=1.18.5
 opencv-python-headless==4.10.0.84
 Pillow>=7.1.2
-pillow-avif-plugin>=1.5.0
 pillow-heif>=0.18.0
 python-dateutil
 python-dotenv

roboflow/util/image_utils.py

@@ -7,6 +7,14 @@
 import yaml
 from PIL import Image
 
+try:
+    import pillow_heif
+
+    pillow_heif.register_heif_opener(thumbnails=False)  # Register for HEIF/HEIC
+    pillow_heif.register_avif_opener()  # Register for AVIF
+except ImportError:
+    pass
+
 
 def check_image_path(image_path):
     """
@@ -74,9 +82,18 @@ def validate_image_path(image_path):
 def file2jpeg(image_path):
     import cv2
 
+    # OpenCV will handle standard formats efficiently
     img = cv2.imread(image_path)
-    image = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
-    pilImage = Image.fromarray(image)
+    if img is not None:
+        # Convert BGR to RGB for PIL
+        image = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
+        pilImage = Image.fromarray(image)
+    else:
+        # If OpenCV fails, the format might be HEIC/AVIF which are handled by PIL
+        pilImage = Image.open(image_path)
+        if pilImage.mode != "RGB":
+            pilImage = pilImage.convert("RGB")
+
     buffered = io.BytesIO()
     pilImage.save(buffered, quality=100, format="JPEG")
     return buffered.getvalue()