Toybrick

标题: 关于在rk3399proD上运行yolov5的官方demo -01 [打印本页]

作者: tiger_k    时间: 2022-3-21 17:51
标题: 关于在rk3399proD上运行yolov5的官方demo -01
首先我按照官方教程刷入了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格式需要注意什么呢?上面的错误该怎么处理呢


作者: tiger_k    时间: 2022-3-21 18:00
https://github.com/rockchip-linu ... es/rknn_yolov5_demo
按照这个进行优化 依然在板子上编译报错
python -m onnxsim yolov5s.onnx  yolov5s.onnx

作者: tiger_k    时间: 2022-7-13 21:09
https://github.com/tiger-k/rk339 ... n/YOLOv5-RK3399ProD




欢迎光临 Toybrick (https://t.rock-chips.com/) Powered by Discuz! X3.3