Typo fix and more flexible COCO for coco-2017

Author: dtronmans

luxonis_ml/data/parsers/classification_directory_parser.py

@@ -53,7 +53,7 @@ def validate_split(split_path: Path) -> dict[str, Any] | None:
                 "valid",
                 "test",
                 "val",
-                "valdation",
+                "validation",
                 "images",
                 "labels",
             }

luxonis_ml/data/parsers/coco_parser.py

@@ -77,12 +77,16 @@ def _is_coco_json(json_path: Path) -> bool:
         try:
             with open(json_path) as f:
                 data = json.load(f)
-            # Distinguish between COCO and FiftyOne classification as they both have labels.json files
-            return (
-                isinstance(data, dict)
-                and "images" in data
-                and "annotations" in data
-                and "categories" in data
+            if not isinstance(data, dict):
+                return False
+            # images is required, annotations is optional (test sets don't have them)
+            if "images" not in data:
+                return False
+            # Categories can be at top level or nested inside info
+            return "categories" in data or (
+                "info" in data
+                and isinstance(data["info"], dict)
+                and "categories" in data["info"]
             )
         except (json.JSONDecodeError, OSError):
             return False

luxonis_ml/data/parsers/luxonis_parser.py

@@ -40,6 +40,7 @@ class ParserType(Enum):
 class LuxonisParser(Generic[T]):
     parsers: dict[DatasetType, type[BaseParser]] = {
         DatasetType.COCO: COCOParser,
+        DatasetType.FIFTYONECLS: FiftyOneClassificationParser,
         DatasetType.VOC: VOCParser,
         DatasetType.DARKNET: DarknetParser,
         DatasetType.YOLOV6: YoloV6Parser,
@@ -53,7 +54,6 @@ class LuxonisParser(Generic[T]):
         DatasetType.YOLOV8BOUNDINGBOX: YOLOv8Parser,
         DatasetType.YOLOV8INSTANCESEGMENTATION: YOLOv8Parser,
         DatasetType.YOLOV8KEYPOINTS: YOLOv8Parser,
-        DatasetType.FIFTYONECLS: FiftyOneClassificationParser,
     }
 
     def __init__(