RV1126笔记十六：吸烟行为检测及部署＜三＞

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

训练并测试(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摄像