大家购买rv1126的开发板, 相信很大程度上希望能使用它的npu做边缘计算, 而不是简单当作一个IPC使用, 当你已经跑过了rknn的几个例程之后, 肯定想试试训练自己的样本,并部署到rv1126.
首先我的训练环境是Windows10+MiniConda,
直接去google一下mini conda, 并安装, 这部分没啥可说的.
打开mini conda的命令行

没有设置环境的情况下, 前面是(base)
创建一个python 3.8的环境并激活
conda create -name py38 python=3.8
conda activate py38
然后从github下载yolov5, 点这里
在conda的命令行下面, 进入yolov5的目录,有个requirements.txt
pip install -r requirements.txt
使用pip安装需要的依赖包.
我想做的demo是一个监控驾驶行为的目标检测系统, 对司机进行驾驶违规行为的判断, 标签class分别为5种
驾驶中, 低头, 打电话, 打哈欠, 闭眼
准备好自己的训练图集, 这里我用了一个简单的usb摄像

本文介绍了基于rv1126开发板使用npu做边缘计算,训练监控驾驶行为目标检测系统的过程。包括搭建Windows10+MiniConda训练环境,下载yolov5并安装依赖,准备训练图集、标记图片,创建训练配置yaml文件,使用pytorch进行训练,最后将训练结果移植到rv1126上。
最低0.47元/天 解锁文章
2443

被折叠的 条评论
为什么被折叠?



