yolov5转成rknn模型

本文主要是介绍yolov5转成rknn模型，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

目录

1 转换方法

2 yolov5版本问题

3 yolov5的输出格式问题

4 转成rknn模型时的格式问题

---

1 转换方法

将yolov5的pth模型转成rknn模型，具体方法分两步

1. 利用yolov5工程中自带的export.py将pt模型转成onnx模型
2. 利用如下脚本生成rknn模型

将onnx模型转化为RV1126平台的rknn模型_cumtchw-CSDN博客

这篇笔记不是为了介绍具体转换方法，而是记录一下中间转换过程中的一些问题。

2 yolov5版本问题

我们想在瑞芯微的板子用C++推理yolov5的rknn模型，目前一般是利用瑞芯微提供的C++ demo

瑞芯微RV1126的YOLOv5官方推理demo

这个demo的后处理针对的三个输出的模型，如果我们想用这个demo，

这篇关于yolov5转成rknn模型的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---