Merge pull request #34 from mgmalana/patch-1

Make EasySet supported_formats not case sensitive
sicara · Apr 4, 2022 · bc60a25 · bc60a25
2 parents 22ffa7f + 7bdfb62
commit bc60a25
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/easyfsl/datasets/easy_set.py b/easyfsl/datasets/easy_set.py
@@ -114,7 +114,7 @@ def list_data_instances(
             class_images = [
                 str(image_path)
                 for image_path in sorted(Path(class_root).glob("*"))
-                if image_path.is_file() & (image_path.suffix in supported_formats)
+                if image_path.is_file() & (image_path.suffix.lower() in supported_formats)
             ]
             images += class_images
             labels += len(class_images) * [class_id]