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是一个吸烟行为检测,先准备好

你可能感兴趣的:(RV1126项目实战,python,深度学习,开发语言)