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

举报
zstar 发表于 2022/08/05 23:26:38 2022/08/05
【摘要】 概述 YOLOv5-5.0版本的四个预训练权重输入的图片尺寸固定为640x640。 但是在YOLOv5-6.1版本,多了几个例如yolov5l6.pt末尾带6的预训练权重,这几个权重是在更大分辨率128...

概述

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

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

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

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

文章来源: zstar.blog.csdn.net,作者:zstar-_,版权归原作者所有,如需转载,请联系作者。

原文链接:zstar.blog.csdn.net/article/details/126117194

【版权声明】本文为华为云社区用户转载文章,如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。