yolov5环境搭建(Windows下cpu训练)

配置前确保自己电脑以下配置完好:

python 最好3.8及以上

notepad++

git

conda(记得添加环境变量到PATH)

pycharm

 

1.下载yolov5的工程文件     https://github.com/ultralytics/yolov5

code-->download zipe913be59cd534384a7a5c6d2eefa74aa.png将其解压到一个不带中文字符的文件夹下(如果带有中文字符,会使OpenCV的cv2.imread()函数读取不了待检测图片或视频

2.更新pip:win+r-->cmd

python -m pip install --upgrade pip

3.创建虚拟环境

切换到d盘  输入d:

conda create -n (自定义虚拟环境的名称)
#我取名为py39,故conda create -n py39

激活虚拟环境

conda activate (虚拟环境名称)

切换到yolov5项目目录下

cd (yolov5工程文件的文件链接)
我的:cd v5

4.安装项目依赖(默认使用cpu计算)

在上一步的cmd窗口上先尝试下面指令:

pip install -r requirements.txt

不出意外会报错,报错原因:无法打开requirements.txt

解决方法:

先执行
pip freeze > requirements.txt
然后再
pip install -r requirements.txt
(或者使用镜像源:pip install -r requirements.txt -i https://pypi.douban.com/simple)

即可成功安装。

5.打开yolov5

找到yolov5文件夹,双击detect.py

b4734d88010644dfb8557eadd51906a9.png

跳转到pycharm,右下角添加解释器,使用conda环境,可执行文件为安装目录下的conda.bat,每个人的conda环境路径不一样,可在  此电脑中全局搜索conda,找到condabin,conda.bat字眼,复制文件地址

使用现有环境,选择刚刚创建的虚拟环境

1bff6f329b054e249fcde4f788d3ec30.png

6.下载权重文件

运行detect.py

报错,指示要去GitHub官网下载权重文件   yolov5s.pt(新版pip会在报错后自动下载该权重文件)

如果不会自动下载,则去官网下载:

eed3b26e188a451d81dbf0e6b1ed8998.png

点击release,下滑找到v5.0,点进去,下载yolov5s.pta4bd3cc9f0034b9dac39d385e2d42251.png将下载好的权重文件放在根目录下,不要放在weigh或者其他文件

1a7793404af34512b70d76aec58a0f84.png

再次运行detect.py,运行成功45af3834de2f4f86a7d7e115446691f1.png

运行结果在runs文件对应的exp中

2e2d19d1cd24495d9800e40c93b19c5c.png

7.测试训练模型

打开train.py,点击运行

提示要下载样本文件,(新版pip会自动进行下载,下载完成后再次运行即可),跳转链接下载,一般为coco128,根据提示创建目录,将下载好的样本文件放入其中,再次运行,开始漫长的等待过程

303b1cf7a3a14907b61b7795e64952a5.png

运行完毕后

73fbdd65390d4f029fe5c1809f9a9a07.png

去runs/train/exp2查看训练结果

d6b637398c144eb18df8e0ae3e7d0b77.png4598c21b604c41fcad9d49470c5dce6a.png6bed2493b03b4d0d8e181828bf1abd57.png

至此,yolov5环境搭建成功

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值