第Y1周：调用YOLOv5官方权重进行检测

• 🍨 本文为🔗365天深度学习训练营 中的学习记录博客
• 🍖 原作者：K同学啊
  

1、前言

YOLOv5分为YOLOv5s、YOLOv5m、YOLOv5l、YOLOv5x四个版本，这里以YOLOv5s为例。

2、下载源码

安装需要的环境：
安装环境依赖包，进入项目文件在文件地址栏中直接输入cmd，打开cmd后输入：pip install -r requirements.txt

3、运行代码

需要的环境：
python>=3.6.0
PyTorch>=1.7

以同样的方法打开cmd，输入命令：python path/to/detect.py --source path/to/img.jpg --weights yolov5s.pt --img 640，注意需要将图片路径替换成你的，例如：python path/to/detect.py --source data\images\zidane.jpg --weights yolov5s.pt --img 640。这里我们使用的是官方已经帮我们训练好的模型。
在这个过程中，你中途可能会遇到这两个问题，不用担心，这是网络问题多试几次就好了：
1、TimeoutError： [WinError 10060] 由于连接方在一段时间后没有正确答复或连接的主机没有反应，连接尝试失败。
2、TypeError: unlink() got an unexpected keyword argument ‘missing_ok’。

如果你还是一直这样，将yolov5s.pt文件放到项目的主目录下就ok了，上面的两个错误都是由于下载这个文件、下载失败缺少这个文件导致的。
命令运行成功后，你将会得到如下的反馈：

进入文件夹查看我们的检测结果：