Yolov5的安装与测试

本文详细介绍如何下载和安装开源目标检测模型Yolov5,涉及源码克隆、预训练模型获取、环境配置及图片、视频和摄像头的实战测试。通过对比不同模型的性能,探讨其在实际应用中的优缺点。
部署运行你感兴趣的模型镜像

1. 下载Yolov5

1.1 下载Yolov5源码

Yolov5 Github地址:https://github.com/ultralytics/yolov5
Github官网下载速度较慢,可以先下载git:https://git-scm.co, 或者使用 镜像源,后者下载速度也会更快些。
具体过程可参考博文:window10安装Git2.33.1

  • 下载完成后,用 win+R 打开cmd输入:
git clone https://github.com/ultralytics/yolov5

若有显示报错:

fatal: unable to access‘https://github.com/ultralytics/yolov5/': 	OpenSSL SSL_read: Connection was reset, errno 10054

解决方法:

git config --global http.sslverify “false”

重新输入安装命令即可直接git clone到yolov5本地工作目录,等待下载完成。
在这里插入图片描述

  • Yolov5代码目录架构:

在这里插入图片描述

1.2 下载Yolov5预训练模型

下载地址:https://github.com/ultralytics/yolov5/releases
在这里插入图片描述

下拉找到最新的release,点开下面的Assets下载Yolov5的四种模型:yolov5s、yolov5m、yolov5l、yolov5x。(.pt文件就是pytorch的模型文件)
在这里插入图片描述
下载后放到Yolov5源码根目录:
在这里插入图片描述

2.安装Yolov5

源码下完后,下面开始安装Yolov5所需模块,首先输入cd进入yolov5的子目录,再输入pip install -r requirements.txt等待安装完成即可。

在这里插入图片描述
显示完成安装。
在这里插入图片描述

3.测试Yolov5

3.1图片测试

  • 通过detect.py对图像进行目标检测,命令如下:
python detect.py --source ./data/images/bus.jpg

此命令是对子目录下/data/imagse里的bus.jpg进行目标检测,原图像bus.jpg如下:
在这里插入图片描述
输入命令执行:
在这里插入图片描述
识别速度为0.287s,最后显示输出结果保存至yolov5的子目录runs/detect/exp10 中:
在这里插入图片描述
目标检测识别图像:
在这里插入图片描述

  • detect.py默认使用同目录下的yolov5s.pt 模型,可以用–weights 进行指定其他模型,如使用yolov5x.pt命令如下:
python detect.py --source ./data/images/bus.jpg --weights yolov5x.pt --conf-thres 0.25

在这里插入图片描述
识别速度为1.609s,检测图像如下:
在这里插入图片描述
对比使用yolov5s.pt模型与yolov5x.pt模型,发现后者识别速度较慢,但是检测效果的准确率大大提升了,这是因为其网络参数更多。

3.2视频测试

首先在data目录下新建文件夹video,再放入一个mp4视频:
在这里插入图片描述

然后通过detect.py对视频进行目标检测

python detect.py --source data/video/v1.mp4

在这里插入图片描述

在这里插入图片描述

输出速度会比较慢,因为视频也是转换为图片放到模型处理的,约每帧216ms,
识别效果如下:
在这里插入图片描述

3.3电脑摄像头测试

使用本机摄像头测试Yolov5实时检测(这里识别的是我的桌面)

python detect.py --source 0

在这里插入图片描述
识别速度约每帧0.26~0.33s。

3.4手机摄像头测试

打开IP摄像头:(IP摄像头支持Android和iOS系统)
在这里插入图片描述

调用命令如下:

python detect.py --source http://admin:admin@10.xxx.xxx.xxx:xxxx  
# @后面填自己的ip地址和端口号

我的局域网地址为http://192.168.43.5:8081,即

 python detect.py --source http://admin:admin@192.168.43.5:8081  

实测效果如下:
在这里插入图片描述

处理速度约每帧0.188s,较电脑摄像头快些。

4. 参考资料

  1. 半小时搞定Yolov5安装配置及使用(详细过程)
  2. 解决git下载很慢的问题
  3. window10安装Git2.33.1
  4. 解决 ERROR: Could not find a version that satisfies the requirement xxx 的问题

您可能感兴趣的与本文相关的镜像

Yolo-v5

Yolo-v5

Yolo

YOLO(You Only Look Once)是一种流行的物体检测和图像分割模型,由华盛顿大学的Joseph Redmon 和Ali Farhadi 开发。 YOLO 于2015 年推出,因其高速和高精度而广受欢迎

评论 11
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值