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

Yolov5的安装与测试

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

Yolov5的安装与测试

目录
  • 1. 下载Yolov5
    • 1.1 下载Yolov5源码
    • 1.2 下载Yolov5预训练模型
  • 2.安装Yolov5
  • 3.测试Yolov5
    • 3.1图片测试
    • 3.2视频测试
    • 3.3电脑摄像头测试
    • 3.4手机摄像头测试
  • 4. 参考资料

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的子目录runsdetectexp10 中:

目标检测识别图像:

  • 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 的问题
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/445388.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

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

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