一、前言
YOLO系列是目前最热门的目标检测算法,YOLOv5分为Yolov5s、Yolov5m、Yolov5l、Yolov5x四个版本,我这里以Yolov5s为例。
二、下载源码
安装需要的环境
● 安装环境依赖包,请进入项目文件在文件地址栏中直接输入cmd,打开cmd后输入:pip install -r requirements.txt,如何打开CMD命令行
三、运行代码
需要的环境
● Python>=3.6.0
● PyTorch>=1.7
以同样的方法打开cmd,输入命令:python path/to/detect.py --source path/to/img.jpg --weights yolov5s.pt --img 640,注意需要将图片路径替换成你的,例如:python detect.py --source data\images\zidane.jpg --weights yolov5s.pt --img 640。这里我们使用的是官方已经帮我们训练好的模型。
在这个过程中,你中途可能会遇到这两个问题,不用担心,这是网络问题多试几次就好了。
● TimeoutError: [WinError 10060] 由于连接方在一段时间后没有正确答复或连接的主机没有反应,连接尝试失败。
● TypeError: unlink() got an unexpected keyword argument ‘missing_ok’
如果你还是一直这样,将YOLOv5s.pt提取码:bum6)文件放到项目的主目录下就ok了,上面的两个错误都是由于下载这个文件、下载失败缺少这个文件导致的。
命令运行成功后,你将会得到如下的反馈:
进行文件夹查看我们的检测结果:
同样的方法打开cmd输入:python detect.py --source MyVideo_2.mp4
我的实验视频(提取码:9vaq),或许你用得到。
你将会得到如下运行结果