Change yolov10 subtype

luxonis · Jun 27, 2024 · e2777c6 · e2777c6
1 parent a34cd4e
commit e2777c6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/yolo/yolov10_exporter.py b/tools/yolo/yolov10_exporter.py
@@ -21,7 +21,7 @@ def __init__(
             model_path,
             imgsz,
             use_rvc2,
-            subtype="yolov8",
+            subtype="yolov10",
             output_names=["output1_yolov10", "output2_yolov10", "output3_yolov10"],
         )
         self.load_model()