Fix/yolov8 output names nn archive (#138)

HonzaCuhel · web-flow · commit a5f82a2c56a9 · 2025-03-20T13:22:52.000+01:00
* Change yolo output names in NN Archive

* Add scipy

* Remove scipy
diff --git a/requirements.txt b/requirements.txt
@@ -5,6 +5,7 @@ PyYAML>=5.3.1
 gcsfs
 luxonis-ml[data,nn_archive,utils]>=0.6.1
 onnx>=1.17.0
+numpy>=1.19.5,<1.27.0
 onnxruntime>=1.20.1
 onnxsim>=0.4.36
 s3fs
diff --git a/tools/yolo/yolov8_exporter.py b/tools/yolo/yolov8_exporter.py
@@ -89,7 +89,7 @@ def __init__(
             imgsz,
             use_rvc2,
             subtype="yolov8",
-            output_names=["output1_yolov8", "output2_yolov8", "output3_yolov8"],
+            output_names=["output1_yolov6r2", "output2_yolov6r2", "output3_yolov6r2"],
         )
         self.load_model()
 

Original file line number	Diff line number	Diff line change
`@@ -89,7 +89,7 @@ def __init__(`
`89`	`89`	`imgsz,`
`90`	`90`	`use_rvc2,`
`91`	`91`	`subtype="yolov8",`
`92`		`- output_names=["output1_yolov8", "output2_yolov8", "output3_yolov8"],`
	`92`	`+ output_names=["output1_yolov6r2", "output2_yolov6r2", "output3_yolov6r2"],`
`93`	`93`	`)`
`94`	`94`	`self.load_model()`
`95`	`95`