作者:殷忆枫

RV1126笔记十四:吸烟行为检测及部署<一>

若该文为原创文章,转载请注明原文出处。

一、项目介绍

最初学习到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上跑了。

如有侵权,或需要完整代码,请及时联系博主。