栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

【目标检测】Anaconda3+PyCharm+Python3.8+PyTorch1.10+Cuda10.2的YOLOv5复现教程

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

【目标检测】Anaconda3+PyCharm+Python3.8+PyTorch1.10+Cuda10.2的YOLOv5复现教程

1 预备工作 1.1 码源下载

下载地址:YOLOv5码源下载
进不去的同学找个梯子哦。

1.2 模型下载

有梯子的同学可以忽略这一步。
没有梯子的话,通过程序下载模型会很慢,所以还是建议通过GitHub官方去下载。
这边也放上下载地址:YOLOv5模型下载
页面向下拉看到下图页面,直接点击,即可下载。

没有梯子的同学,需要的话留言,我会放上百度云连接。

1.3 环境配置

我用的是Anaconda虚拟环境,具体环境配置我会单独写一篇文章。
将下载下来的yolov5-master.zip文件解压。


接下来就是配置环境了。
打开Anaconda prompt输入如下代码,进入虚拟环境。这边的虚拟环境是自定义的虚拟环境,activate后面加的是你自定义的虚拟环境。

activate pytorch


打开程序可以看到有一个requirements.txt文件,我们通过这个文件来安装没有的包。

在终端虚拟环境中输入如下代码安装requirements.txt中所需要的包。

pip install -r requirements.txt

这边要注意一点,一定去找到requirements.txt当前所在的文件夹去安装,不然会出错。
ERROR: Could not open requirements file: [Errno 2] No such file or directory: ‘requirements.txt’

按照如下操作通过终端进入解压后的文件夹。
博主是将压缩包解压在D盘,因此输入

d:

进入D盘。

点击解压后的文件夹,查看属性,这一步主要是要查看文件夹所属路径。如果文件放得比较浅,可以直接敲。

在终端输入进入文件夹

cd Downloadyolov5-master

cd后面跟的就是你的解压后文件夹路径,这里就是博主所放的一个路径。

接着就可以安装requirements.txt中所需要的包啦。
输入指令

pip install -r requirements.txt


没有出现错误就是安装成功啦。

2 YOLOv5预测

在PyCharm平台上打开detect.py,直接点击运行,看看有没有出错,博主当时下载完模型后是没有出错的,在yolov5-masterdataimages文件夹中有两张照片,你也可以放上在自己的照片,进行预测。
预测后的结果在yolov5-masterrunsdetectexp8文件夹中,里面有检测物体方框,物体类别和预测的置信度。如图所示。

预测后图片。

当然,如果你想提高检测的置信度,可以改变模型,将原来

parser.add_argument('--weights', nargs='+', type=str, default=ROOT / 'yolov5s.pt', help='model path(s)')

中的模型改成其他的模型,这里的模型从官网下载后的模型。
例如,改成yolo5m.pt这个模型,将原代码改成

parser.add_argument('--weights', nargs='+', type=str, default=ROOT / 'yolov5m.pt', help='model path(s)')

即可。
可以看到检测结果置信度有明显提升,当然,模型越大检测的速度就会越慢。

累了,明天继续train的部分
如果有没有gpu的朋友,还会出Google colab远程服务器的教程,看心情写,进行就先这样。
detect.py文件运行有错误的可以留言一起讨论哦~

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/580427.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号