当前位置:   article > 正文

YOLOV5从模型训练到部署在NPU上实现流程_yolov5使用npu

yolov5使用npu

        踩了三天坑,看了各种资料,跑官网提供的C++ demo,python demo可以得到正确的结果,但是从头从 pt->onnx->rknn文件总是无法成功。因为不好的工作思路,习惯找开源,于是到git 上下载各种版本代码尝试都出推理结果不正确的问题。后来头疼的想要放弃,后来看到博客给提供了思路,分别为:

rknn导出yolov5模型在python中可正常推理,在c++中无结果._rknn 模型推理无法获取结果-CSDN博客Yolov5笔记--RKNN推理部署源码的粗略理解_rknn yolov5-CSDN博客

        废话不多说,进入正题,给大家提供一套完整的从训练到部署的使用流程。

训练代码链接,使用官网提供的即可:

https://www.wpsshop.cn/w/代码探险家/article/detail/953289

推荐阅读
相关标签
  

闽ICP备14008679号