关于在rk3399proD上运行yolov5的官方demo -01

[复制链接] · 发表于 2022-3-21 17:51:27

首先我按照官方教程刷入了debian10系统，在测试yolov3示例【新提醒】Toybrick-开源社区-人工智能-人工智能开发系列(3) YOLOV3开发与实现 (rock-chips.com)

运行成功
接着开始测试onnx目录下的demo，此处需要升级rknn到1.7.1
先用官方demo中的yolov5s.onnx试试编译推理成功

然后下载官方的yolov5s.pt 测试模型转换和编译（用的是yolov5 V6.0版本）

然后我测试了自己导出的onnx文件进行编译和推理

问题：对于onnx版本的yolov5 demo 从官方下载的yolov5sV6.0版本的yolov5s.pt 文件转换成onnx格式需要注意什么呢？上面的错误该怎么处理呢

只看该作者 · 发表于 2022-3-21 18:00:16

https://github.com/rockchip-linu ... es/rknn_yolov5_demo
按照这个进行优化依然在板子上编译报错
python -m onnxsim yolov5s.onnx yolov5s.onnx

只看该作者 · 发表于 2022-7-13 21:09:46