Add support for YOLOv9 - 11

src/labelformat/cli/cli.py

@@ -21,7 +21,7 @@ def main() -> None:
 The CLI interface is available as the `labelformat` command.
 
 Supported label formats for object detection:
-- YOLOv5, YOLOv6, YOLOv7, YOLOv8
+- YOLOv5, YOLOv6, YOLOv7, YOLOv8, YOLOv9, YOLOv10, YOLOv11
 - COCO
 - VOC
 - Labelbox

src/labelformat/formats/__init__.py

@@ -35,3 +35,15 @@
     YOLOv8ObjectDetectionInput,
     YOLOv8ObjectDetectionOutput,
 )
+from labelformat.formats.yolov9 import (
+    YOLOv9ObjectDetectionInput,
+    YOLOv9ObjectDetectionOutput,
+)
+from labelformat.formats.yolov10 import (
+    YOLOv10ObjectDetectionInput,
+    YOLOv10ObjectDetectionOutput,
+)
+from labelformat.formats.yolov11 import (
+    YOLOv11ObjectDetectionInput,
+    YOLOv11ObjectDetectionOutput,
+)

src/labelformat/formats/yolov10.py

@@ -0,0 +1,17 @@
+from labelformat.cli.registry import Task, cli_register
+
+from .yolov8 import YOLOv8ObjectDetectionInput, YOLOv8ObjectDetectionOutput
+
+"""
+YOLOv10 format follows the same specs as YOLOv8.
+"""
+
+
+@cli_register(format="yolov10", task=Task.OBJECT_DETECTION)
+class YOLOv10ObjectDetectionInput(YOLOv8ObjectDetectionInput):
+    pass
+
+
+@cli_register(format="yolov10", task=Task.OBJECT_DETECTION)
+class YOLOv10ObjectDetectionOutput(YOLOv8ObjectDetectionOutput):
+    pass

src/labelformat/formats/yolov11.py

@@ -0,0 +1,17 @@
+from labelformat.cli.registry import Task, cli_register
+
+from .yolov8 import YOLOv8ObjectDetectionInput, YOLOv8ObjectDetectionOutput
+
+"""
+YOLOv11 format follows the same specs as YOLOv8.
+"""
+
+
+@cli_register(format="yolov11", task=Task.OBJECT_DETECTION)
+class YOLOv11ObjectDetectionInput(YOLOv8ObjectDetectionInput):
+    pass
+
+
+@cli_register(format="yolov11", task=Task.OBJECT_DETECTION)
+class YOLOv11ObjectDetectionOutput(YOLOv8ObjectDetectionOutput):
+    pass

src/labelformat/formats/yolov9.py

@@ -0,0 +1,17 @@
+from labelformat.cli.registry import Task, cli_register
+
+from .yolov8 import YOLOv8ObjectDetectionInput, YOLOv8ObjectDetectionOutput
+
+"""
+YOLOv9 format follows the same specs as YOLOv8.
+"""
+
+
+@cli_register(format="yolov9", task=Task.OBJECT_DETECTION)
+class YOLOv9ObjectDetectionInput(YOLOv8ObjectDetectionInput):
+    pass
+
+
+@cli_register(format="yolov9", task=Task.OBJECT_DETECTION)
+class YOLOv9ObjectDetectionOutput(YOLOv8ObjectDetectionOutput):
+    pass