【目标检测】YOLOv5:640与1280分辨率效果对比

概述

YOLOv5-5.0版本的四个预训练权重输入的图片尺寸固定为640x640。
但是在YOLOv5-6.1版本,多了几个例如yolov5l6.pt末尾带6的预训练权重,这几个权重是在更大分辨率1280x1280进行训练的。
因此我想在自己的电脑上测试一下相同的数据集从640x640到1280x1280效果会有多少提升。
原本是想在VisDrone数据集上进行对比测试,然后当换用更大分辨率之后,即使batch_size减为1,依然面临显存不足的问题。
于是我使用了自己的数据集,采用yolov5l模型,跑了100个epoch,mAP效果如下表所示:

算法mAP@.5mAP@.5:.95
yolov5(640)95.6%67.6%
yolov5(1280)98.7%75.7%

注:自己的数据集样本量不多,并且目标较稀疏,因此mAP比较高。

从表中可以看出,换用更大尺寸的分辨率之后,mAP指标还是有明显提升的。
下面是Result.png
yolov5(640):
请添加图片描述
yolov5(1280):
请添加图片描述
如图所示,可以看到yolov5(1280)的收敛速度更快一些。

评论 55
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

zstar-_

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值