【目标检测】超分重建对小目标检测有效性探究

举报
zstar 发表于 2022/08/27 23:11:37 2022/08/27
【摘要】 前言 前段时间在看一些小目标检测的文献综述,看到一种思路是将图片超分重建之后再送到检测网络中进行检测。 优点是检测效果提升,缺点是可能会造成伪影。 实践出真知,本篇就用DOTA-test数据集中的一张图...

前言

前段时间在看一些小目标检测的文献综述,看到一种思路是将图片超分重建之后再送到检测网络中进行检测。
优点是检测效果提升,缺点是可能会造成伪影。
实践出真知,本篇就用DOTA-test数据集中的一张图像来做测试,探究超分重建是否对小目标检测起到帮助。

超分模型

本次实验选用的是之前使用过的Real-ESRGAN,作者提供了一个方便快捷的工具包,可以用命令行的方式快速进行图像/视频的超分转换。

下面这两张图是上篇博文用到的大图局部。

转换之前:
在这里插入图片描述
在这里插入图片描述

转换之后:
在这里插入图片描述
在这里插入图片描述
可以看到图像的清晰度提升了不少,不过由于该模型的预训练权重作者主要是在动漫图上进行训练,因此重建出的结果会一些动漫风格。

结果对比

由于重建出的图片默认会是原图的4倍大小,为了对比的公平性,将重建之后的原图用线性插值方式(PS自动调节),重设为原图尺寸。

然后分别输入YOLOv5网络中,利用之前训练好的YOLOv5L模型进行检测,其结果如下图所示:

重建前:
在这里插入图片描述
重建后:
在这里插入图片描述
从这组对比可以发现,重建之后确实检测出了更多图像。

下面再换一块区域,进行对比:

重建前:
在这里插入图片描述
重建后:
在这里插入图片描述
从这组对比可以看到,重建之后,模型检测出了原本没检测到的面包车(蓝色框),但与此同时,也丢失了部分原本检测出的对象。

总结

超分重建相当于是对图像进行了优化,微调了整体的光照色彩,减少了噪点。但这并不代表一定能优化目标的检测,最多是从另一个视角进行再次检测,效果变好变坏都有可能。

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

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

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

评论(0

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

全部回复

上滑加载中

设置昵称

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

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

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