search

cqu20160901 / yolov9_onnx_tensorRT_rknn_horizon

yolov9 部署，便于移植不同平台（onnx、tensorRT、rknn、Horizon），全网首个通用部署示例。
11 stars 1 forks source link

export to onnx error #1

Open HduHestin opened 3 months ago

HduHestin commented 3 months ago

你好，我参考这篇文章进行yolov9-c.pt转rknn的步骤。 https://blog.csdn.net/zhangqian_1/article/details/136321979

其中common.py 和 yolo.py 和 detect.py中都按照文章所描述的添加了。

然后我使用如下的命令： python detect.py --source './data/images/horses.jpg' --img 640 --device cpu --weights './ptsmodels/yolov9-c.pt' --name yolov9_c_c_640_detect image 然后利用https://github.com/cqu20160901/yolov9_onnx_tensorRT_rknn_horizon/blob/main/yolov9_onnx/yolov9s_onnx_demo_zq.py 中的方法对onnx的文件进行测试 原图： image 结果： image

希望您能给我一些帮助。

cqu20160901 commented 2 months ago

问题解决了吗？

HduHestin commented 2 months ago

解决了，我更改结构的时候改错Class了，应该是在DualDDetect类中修改 .我改成DualDetect类了