adds support for heic/avif too

joaomarcoscrs · joaomarcoscrs · commit 9ba184bc8346 · 2025-03-13T18:45:40.000Z
diff --git a/requirements.txt b/requirements.txt
@@ -6,6 +6,8 @@ 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
 requests
diff --git a/roboflow/core/project.py b/roboflow/core/project.py
@@ -24,6 +24,8 @@
     "image/webp",
     "image/tiff",
     "image/tif",
+    "image/avif",
+    "image/heic",
 }
 
 
diff --git a/roboflow/util/folderparser.py b/roboflow/util/folderparser.py
@@ -7,7 +7,7 @@
 
 from .image_utils import load_labelmap
 
-IMAGE_EXTENSIONS = {".jpg", ".jpeg", ".png", ".bmp", ".tiff", ".tif"}
+IMAGE_EXTENSIONS = {".jpg", ".jpeg", ".png", ".bmp", ".tiff", ".tif", ".avif", ".heic"}
 ANNOTATION_EXTENSIONS = {".txt", ".json", ".xml", ".csv", ".jsonl"}
 LABELMAPS_EXTENSIONS = {".labels", ".yaml", ".yml"}
 
diff --git a/tests/images/whatsnew.avif b/tests/images/whatsnew.avif
diff --git a/tests/test_project.py b/tests/test_project.py
@@ -16,6 +16,7 @@ def test_check_valid_image_with_accepted_formats(self):
             "hand-rabbit.PNG",
             "woodland-rabbit.png",
             "file_example_TIFF_1MB.tiff",
+            "sky-rabbit.heic",
         ]
 
         for image in images_to_test:
@@ -24,7 +25,6 @@ def test_check_valid_image_with_accepted_formats(self):
     def test_check_valid_image_with_unaccepted_formats(self):
         images_to_test = [
             "sky-rabbit.gif",
-            "sky-rabbit.heic",
         ]
 
         for image in images_to_test:

Original file line number	Diff line number	Diff line change
`@@ -24,6 +24,8 @@`
`24`	`24`	`"image/webp",`
`25`	`25`	`"image/tiff",`
`26`	`26`	`"image/tif",`
	`27`	`+ "image/avif",`
	`28`	`+ "image/heic",`
`27`	`29`	`}`
`28`	`30`
`29`	`31`
Original file line number	Diff line number	Diff line change
`@@ -16,6 +16,7 @@ def test_check_valid_image_with_accepted_formats(self):`
`16`	`16`	`"hand-rabbit.PNG",`
`17`	`17`	`"woodland-rabbit.png",`
`18`	`18`	`"file_example_TIFF_1MB.tiff",`
	`19`	`+ "sky-rabbit.heic",`
`19`	`20`	`]`
`20`	`21`
`21`	`22`	`for image in images_to_test:`
`@@ -24,7 +25,6 @@ def test_check_valid_image_with_accepted_formats(self):`
`24`	`25`	`def test_check_valid_image_with_unaccepted_formats(self):`
`25`	`26`	`images_to_test = [`
`26`	`27`	`"sky-rabbit.gif",`
`27`		`- "sky-rabbit.heic",`
`28`	`28`	`]`
`29`	`29`
`30`	`30`	`for image in images_to_test:`