【目标检测】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 Download\yolov5-master

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

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

pip install -r requirements.txt

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

2 YOLOv5预测

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

预测后图片。

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

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文件运行有错误的可以留言一起讨论哦~