若该文为原创文章,转载请注明原文出处。
训练并测试(windows)
一、yolov5安装
1、下载rk优化后的yolov5
git clone https://github.com/airockchip/yolov5.git
下载后,我是放到E:\miniconda3\envs目录下,miniconda3是安装miniconda的目录。可以放到其他地方,后续操作需要自行修改。
2、安装依赖
打开我们下载好的yolov5文件夹,打开conda 命令行进入该目录,注意要从base环境切换到yolo环境。
cd yolov5-master_rk
pip install -r requirements.txt // 下载相当慢,使用下面国内源
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
3、测试
安装完成后,输入下面命令测试,可以在run下的detect看到结果。
python detect.py --source data/images/zidane.jpg --weights yolov5s.pt
二、准本训练图集
我想做的demo是一个吸烟行为检测,先准备好自己的训练图集,可以网上下面,也可以使用USB摄像