训练自己的yolov5样本, 并部署到rv1126 <一>

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

大家购买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摄像

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值