Update imagenet_format.py (cvat-ai#152)

Added support of uppercase file name extension in ImageNet format
TOsmanov · Mar 9, 2021 · d6902a8 · d6902a8
1 parent e32a4f9
commit d6902a8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/datumaro/plugins/imagenet_format.py b/datumaro/plugins/imagenet_format.py
@@ -40,7 +40,7 @@ def _load_items(self, path):
         items = {}
         for image_path in glob(osp.join(path, '*', '*')):
             if not osp.isfile(image_path) or \
-                    osp.splitext(image_path)[-1] not in \
+                    osp.splitext(image_path)[-1].lower() not in \
                         ImagenetPath.IMAGE_EXT_FORMATS:
                 continue
             label = osp.basename(osp.dirname(image_path))