Edit subtype to yolov6r2 (#137)

HonzaCuhel · web-flow · commit 94a369a0a69c · 2025-03-05T09:16:32.000+01:00
diff --git a/tools/yolo/yolov6_exporter.py b/tools/yolo/yolov6_exporter.py
@@ -29,7 +29,7 @@ def __init__(
             model_path,
             imgsz,
             use_rvc2,
-            subtype="yolov6",
+            subtype="yolov6r2",
             output_names=["output1_yolov6r2", "output2_yolov6r2", "output3_yolov6r2"],
         )
         self.load_model()
diff --git a/tools/yolov6r3/gold_yolo_exporter.py b/tools/yolov6r3/gold_yolo_exporter.py
@@ -33,7 +33,7 @@ def __init__(
             model_path,
             imgsz,
             use_rvc2,
-            subtype="yolov6",
+            subtype="yolov6r2",
             output_names=["output1_yolov6r2", "output2_yolov6r2", "output3_yolov6r2"],
         )
         self.load_model()
diff --git a/tools/yolov6r3/yolov6_r3_exporter.py b/tools/yolov6r3/yolov6_r3_exporter.py
@@ -40,7 +40,7 @@ def __init__(
             model_path,
             imgsz,
             use_rvc2,
-            subtype="yolov6",
+            subtype="yolov6r2",
             output_names=["output1_yolov6r2", "output2_yolov6r2", "output3_yolov6r2"],
         )
         self.load_model()

Original file line number	Diff line number	Diff line change
`@@ -29,7 +29,7 @@ def __init__(`
`29`	`29`	`model_path,`
`30`	`30`	`imgsz,`
`31`	`31`	`use_rvc2,`
`32`		`- subtype="yolov6",`
	`32`	`+ subtype="yolov6r2",`
`33`	`33`	`output_names=["output1_yolov6r2", "output2_yolov6r2", "output3_yolov6r2"],`
`34`	`34`	`)`
`35`	`35`	`self.load_model()`
Original file line number	Diff line number	Diff line change
`@@ -33,7 +33,7 @@ def __init__(`
`33`	`33`	`model_path,`
`34`	`34`	`imgsz,`
`35`	`35`	`use_rvc2,`
`36`		`- subtype="yolov6",`
	`36`	`+ subtype="yolov6r2",`
`37`	`37`	`output_names=["output1_yolov6r2", "output2_yolov6r2", "output3_yolov6r2"],`
`38`	`38`	`)`
`39`	`39`	`self.load_model()`
Original file line number	Diff line number	Diff line change
`@@ -40,7 +40,7 @@ def __init__(`
`40`	`40`	`model_path,`
`41`	`41`	`imgsz,`
`42`	`42`	`use_rvc2,`
`43`		`- subtype="yolov6",`
	`43`	`+ subtype="yolov6r2",`
`44`	`44`	`output_names=["output1_yolov6r2", "output2_yolov6r2", "output3_yolov6r2"],`
`45`	`45`	`)`
`46`	`46`	`self.load_model()`