Merge pull request #431 from roboflow/feature/yolov10

Add YOLOv10 Object Detection Support
roboflow · May 25, 2024 · 39ebdad · 39ebdad
2 parents b6af7a7 + 3b5eaeb
commit 39ebdad
Show file tree

Hide file tree

Showing 6 changed files with 731 additions and 1 deletion.
diff --git a/inference/core/version.py b/inference/core/version.py
@@ -1,4 +1,4 @@
-__version__ = "0.11.1"
+__version__ = "0.11.2"
 
 
 if __name__ == "__main__":

diff --git a/inference/models/__init__.py b/inference/models/__init__.py
@@ -50,3 +50,4 @@
     YOLOv8ObjectDetection,
 )
 from inference.models.yolov9 import YOLOv9ObjectDetection
+from inference.models.yolov10 import YOLOv10ObjectDetection
diff --git a/inference/models/utils.py b/inference/models/utils.py
@@ -20,6 +20,7 @@
     YOLOv8InstanceSegmentation,
     YOLOv8ObjectDetection,
     YOLOv9ObjectDetection,
+    YOLOv10ObjectDetection,
 )
 from inference.models.yolov8.yolov8_keypoints_detection import YOLOv8KeypointsDetection
 
@@ -50,6 +51,13 @@
     ("object-detection", "yolo_nas_s"): YOLONASObjectDetection,
     ("object-detection", "yolo_nas_m"): YOLONASObjectDetection,
     ("object-detection", "yolo_nas_l"): YOLONASObjectDetection,
+    ("object-detection", "yolov10"): YOLOv10ObjectDetection,
+    ("object-detection", "yolov10s"): YOLOv10ObjectDetection,
+    ("object-detection", "yolov10n"): YOLOv10ObjectDetection,
+    ("object-detection", "yolov10b"): YOLOv10ObjectDetection,
+    ("object-detection", "yolov10m"): YOLOv10ObjectDetection,
+    ("object-detection", "yolov10l"): YOLOv10ObjectDetection,
+    ("object-detection", "yolov10x"): YOLOv10ObjectDetection,
     ("instance-segmentation", "stub"): InstanceSegmentationModelStub,
     (
         "instance-segmentation",