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 的问题
  • 8
    点赞
  • 91
    收藏
    觉得还不错? 一键收藏
  • 11
    评论
YOLOv5是一种用于目标检测的深度学习模型,它是YOLO系列的最新版本,能够在速度和准确性方面实现更好的平衡。下面是YOLOv5安装和使用步骤: 1. 安装Python和PyTorch YOLOv5需要Python 3.8或更高版本以及PyTorch 1.7或更高版本。你可以通过以下命令安装: ``` pip install torch torchvision torchaudio ``` 2. 下载YOLOv5 YOLOv5的代码和预训练模型可以从GitHub上下载。你可以使用以下命令克隆代码仓库: ``` git clone https://github.com/ultralytics/yolov5.git ``` 3. 数据集准备 在训练YOLOv5之前,必须准备好数据集。YOLOv5支持多种数据格式,包括COCO、Pascal VOC和YOLO格式。你可以使用自己的数据集或使用公共数据集。 4. 训练YOLOv5 在训练YOLOv5之前,需要修改相应的配置文件。YOLOv5提供了多个配置文件,你可以选择适合自己的配置文件。配置文件的路径是`yolov5/models/yolov5{version}.yaml`。修改完成后,可以使用以下命令开始训练: ``` python train.py --img 640 --batch 16 --epochs 50 --data /path/to/data.yaml --cfg /path/to/yolov5{version}.yaml --weights /path/to/weights ``` 其中,`img`指定图像的大小,`batch`指定批次大小,`epochs`指定训练轮数,`data`指定数据集配置文件的路径,`cfg`指定模型配置文件的路径,`weights`指定预训练模型的路径。训练完成后,模型权重保存在`runs/train/exp/weights/best.pt`。 5. 测试YOLOv5 可以使用以下命令测试YOLOv5: ``` python detect.py --source /path/to/image/or/video --weights /path/to/weights --conf 0.25 ``` 其中,`source`指定输入图像或视频的路径,`weights`指定模型权重的路径,`conf`指定置信度阈值。运行完成后,检测结果保存在`runs/detect/exp`目录下。 以上就是YOLOv5安装和使用步骤。
评论 11
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值