Support for UltralyticsParser. (#367)

Author: kkeroo

luxonis_ml/data/README.md

@@ -416,6 +416,44 @@ The supported formats are:
             └── test/
     ```
 
+- [**YOLOv8-v12**](https://roboflow.com/formats/yolov8-pytorch-txt) and [**Ultralytics**](https://docs.ultralytics.com/datasets/)
+
+  - Roboflow format (supports YOLOv8-v12)
+    ```plaintext
+        dataset_dir/
+        ├── train/
+        │   ├── images/
+        │   │   ├── img1.jpg
+        │   │   ├── img2.jpg
+        │   │   └── ...
+        │   ├── labels/
+        │   │   ├── img1.txt
+        │   │   ├── img2.txt
+        │   │   └── ...
+        ├── valid/
+        ├── test/
+        └── *.yaml
+    ```
+  - Ultralytics format
+    ```plaintext
+        dataset_dir/
+        ├── images/
+        │   ├── train/
+        │   │   ├── img1.jpg
+        │   │   ├── img2.jpg
+        │   │   └── ...
+        │   ├── val/
+        │   └── test/
+        ├── labels/
+        │   ├── train/
+        │   │   ├── img1.txt
+        │   │   ├── img2.txt
+        │   │   └── ...
+        │   ├── val/
+        │   └── test/
+        └── *.yaml
+    ```
+
 - [**Pascal VOC XML**](https://roboflow.com/formats/pascal-voc-xml)
 
   ```plaintext

luxonis_ml/data/parsers/__init__.py

@@ -10,6 +10,7 @@
 from .voc_parser import VOCParser
 from .yolov4_parser import YoloV4Parser
 from .yolov6_parser import YoloV6Parser
+from .yolov8_parser import YOLOv8Parser
 
 __all__ = [
     "BaseParser",
@@ -22,6 +23,7 @@
     "SegmentationMaskDirectoryParser",
     "TensorflowCSVParser",
     "VOCParser",
+    "YOLOv8Parser",
     "YoloV4Parser",
     "YoloV6Parser",
 ]

luxonis_ml/data/parsers/classification_directory_parser.py

@@ -33,7 +33,8 @@ def validate_split(split_path: Path) -> dict[str, Any] | None:
         classes = [
             d
             for d in split_path.iterdir()
-            if d.is_dir() and d.name not in {"train", "valid", "test"}
+            if d.is_dir()
+            and d.name not in {"train", "valid", "test", "images", "labels"}
         ]
         if not classes:
             return None

luxonis_ml/data/parsers/luxonis_parser.py

@@ -25,6 +25,7 @@
 from .voc_parser import VOCParser
 from .yolov4_parser import YoloV4Parser
 from .yolov6_parser import YoloV6Parser
+from .yolov8_parser import YOLOv8Parser
 
 
 class ParserType(Enum):
@@ -48,6 +49,7 @@ class LuxonisParser(Generic[T]):
         DatasetType.SEGMASK: SegmentationMaskDirectoryParser,
         DatasetType.SOLO: SOLOParser,
         DatasetType.NATIVE: NativeParser,
+        DatasetType.YOLOV8: YOLOv8Parser,
     }
 
     def __init__(