若该文为原创文章,转载请注明原文出处。
一、项目介绍
最初学习到RV1126的RKNN时,就想使用yolov5训练吸烟行为,并部署到RV1126,实现AI功能。
也可以训练安全冒,打哈欠,接打电话等行为。
RV1126的RKNN自带AI功能,可以实现物体检测等功能。
正点原子的RV1126资料有详细的教程,不做过多介绍。
最后部署到RV1126,这是推荐正点原子的RV1126开发板,易百纳开发板亲测无法跑起来,其它开发板需要自行测试。
二、整体方案介绍
1、搭建yolov5训练环境(windows)
2、训练并测试(windows)
3、转换成onnx模型(windows)
4、ubuntu16.04上搭建转化成RKNN环境(ubuntu16.04)
5、onnx模型和rknn模型仿真测试
6、demo移植测试(ubuntu16.04)
7、实时采集数据并屏幕显示(开发板)
实现方式: 使用yolov5训练,生成pt文件,把pt文件转成onnx,在把onnx转成rknn模型,之后就可以在RV1126上跑了。
如有侵权,或需要完整代码,请及时联系博主。