yolov5 rknn
时间: 2023-09-20 18:06:36 浏览: 80
yolov5 rknn是指将yolov5模型转换为适用于RK1808硬件的模型的过程。在转换过程中,需要使用rknn-toolkit 1.7.1进行模型转换和验证,并且需要将pt模型转化为RK可接受的模型。目前硬件环境RK仅能解析yolov5 5.*版本的模型,对于yolov5更新后的6.0版本,硬件并不支持。因此,需要使用rknn_model_zoo来实现对yolov5模型的转化。转换流程包括训练pt模型best.pt,将其转换为onnx模型best.onnx,并进行模型推理。在转换过程中,如果出现报错,可以参考博文https://blog.csdn.net/magicnnonn/article/details/124343085对common.py文件进行修改。请注意,如果训练尺寸不是640,anchors会自动聚类重新生成。<span class="em">1</span><span class="em">2</span><span class="em">3</span>
#### 引用[.reference_title]
- *1* *2* [关于yolov5模型转换为rknn](https://blog.csdn.net/weixin_44220689/article/details/127110943)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"]
- *3* [yolov5_for_rknn:PyTorch中的YOLOv5> ONNX> RKNN](https://download.csdn.net/download/weixin_42116847/18557073)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"]
[ .reference_list ]