新闻中心

改进的检测算法:用于高分辨率光学遥感图像目标检测

2024-06-05
浏览次数:
返回列表

01 前景概要

目前,难以在检测效率和检测结果之间取得适当的平衡。我们就研究出了一种用于高分辨率光学遥感图像中目标检测的增强YOLOv5算法,利用多层特征金字塔、多检测头策略和混合注意力模块来提高光学遥感图像的目标检测网络的效果。根据SIMD数据集,新算法的mAP比YOLOv5好2.2%,比YOLOX好8.48%,在检测结果和速度之间实现了更好的平衡。

02 背景&动机

随着远感技术的快速发展,高分辨率光学远感图像已被用于描述地球表面的许多物体,包括飞机、汽车、建筑物等。目标检测在远感图像的解释中起着至关重要的作用,可用于远感图像的分割、描述和目标跟踪。然而,由于其相对较大的视场和高海拔的必要性,航空光学远感图像表现出尺度、视点特异性、随机定向和高背景复杂性的多样性,而大多数传统数据集包含地面视图。因此,用于构建人工特征检测技术传统上在精度和速度方面具有较大差异的记录。由于社会的需求和深度学习发展的支持,在光学远感图像中使用神经网络进行目标检测是必要的。

目前结合深度学习分析光学远感照片的目标检测算法可以分为有监督、无监督和弱监督三种。然而,由于无监督和弱监督算法的复杂性和不确定性,监督算法是最常用的算法。此外,监督目标检测算法可分为单阶段或两阶段。基于飞机通常位于机场,船舶通常位于港口和海洋的假设,在下采样星图像中检测机场和港口,然后将发现的目标映射回原始的超高分辨率卫星图像,可以同时检测不同大小的物体。有研究者提出了一种基于RCNN的旋转目标检测方法,通过解决目标方向的随机化问题,提高遥感图像中目标检测的准确性。

03 新算法研究

目前的大多数YOLO系列检测头都是基于FPN和PAFPN的输出特性,其中基于FPN的网络,如YOLOv3,及其变体如下图a所示,它们直接利用单向融合特性进行输出。而基于PAFPN算法的YOLOv4和YOLOv5在此基础上增加了一个低电平到高电平的信道,它直接向上传输低级信号(下图b)。

☞☞☞AI 智能聊天, 问答助手, AI 智能搜索, 免费无限量使用 DeepSeek R1 模型☜☜☜

改进的检测算法:用于高分辨率光学遥感图像目标检测

如上图所示,在一些研究中,TPH-YOLOv5模型中的特定检测任务添加了一个检测头。在上图b和c中,只有PAFPN功能可以用于输出,而FPN功能未得到充分利用。因此,YOLOv7将三个辅助磁头连接到FPN输出,如上图d所示,尽管辅助磁头仅用于“粗粮选择”,并且重量评估较低。SSD的检测头是为了改善YOLO网络对锚集过于粗糙的设计而提出的,并提出了基于多尺度的密集锚设计组成。如图f所示,该策略可以同时利用PANet和FPN的特征信息。此外,还有64倍下采样处理直接添加了输出,这使得网络包含先前的全局信息。

多检测头方法可以有效地利用网络的输出特征。改进的YOLO是一个用于高分辨率遥感照片的目标检测网络。如下图所示:

改进的检测算法:用于高分辨率光学遥感图像目标检测

主干网的基本结构是以C3和卷积模块为核心的CSP密集网。在数据增强之后,图像被馈送到网络中,并且在核大小为6的Conv模块进行信道混合之后,许多卷积模块进行特征检索。在名为SPPF的特征增强模块之后,它们连接到Neck的PANet。为了提高网络的检测能力,进行了双向特征融合。Conv2d用于独立扩放融合的特征层,以生成多层输出。如下图所示,NMS算法将所有单层检测器的输出组合在一起,生成最终的检测帧。

改进的检测算法:用于高分辨率光学遥感图像目标检测

下图b描述了改进的YOLO网络每个模块的结构组成。

改进的检测算法:用于高分辨率光学遥感图像目标检测

Conv包括2D卷积层、BN层批量归一化和Silu激活函数,C3包括两个2D卷积层和一个瓶颈层,Upsample是上采样层。SPPF模块是SPP模块的加速版本,MAB模块如上面提及的所示,其中ECA如左下角所示。在没有降维的信道级全局平均池化之后,考虑到每个信道与其k个邻居的关系,使用大小为k的快速1D卷积来捕获局部跨信道交互信息,从而有效地执行ECA。上述两个变换沿着两个空间方向收集特征,以产生一对方向感知特征图,然后将其连接起来,并使用卷积和Sigmoid函数进行修改,以提供注意力输出。

04 实验及可视化

SIMD数据集是一个多类别、开源、高分辨率遥感对象检测数据集,共包含15个类别,如图4所示。此外,SIMD数据集更多地分布在中小型目标(w

改进的检测算法:用于高分辨率光学遥感图像目标检测

改进的检测算法:用于高分辨率光学遥感图像目标检测

可以将SPPF模块的输出连接到输出头,从而识别图像中的大目标。然而,SPPF模块的输出具有多个连接,并且涉及多个尺度的目标,因此直接将其用于探测头来识别大型物体将导致较差的模型表示,如上图所示,显示了添加MAB模块前后一些检测结果热图的视觉比较。在添加MAB模块后,该检测头专注于检测大目标,而将小目标的预测分配给其他预测头,提高了模型的表达效果,也更符合YOLO算法中基于目标大小划分检测头的要求。

改进的检测算法:用于高分辨率光学遥感图像目标检测

一些检测结果如上图所示。从各个检测结果来看,与其他算法没有太大差异,但与其他算法相比,我们研究的算法在保证时间消耗不显著增加的同时,提高了模型的检测效果,并利用注意力机制增强了模型的表达效果。

改进的检测算法:用于高分辨率光学遥感图像目标检测

以上就是改进的检测算法:用于高分辨率光学遥感图像目标检测的详细内容,更多请关注其它相关文章!


# 算法  # 检测  # 将其  # 多个  # 出了  # 如下图  # 如上图  # 开源  # 连接到  # 是一个  # 信道  # 所示  # 网站首页seo设置  # 网站怎么上广告推广  # 天府新区网站建设推广  # 新媒体营销推广市场  # SEO劫持教学  # 化工网站建设哪家售后好  # seo算法入门教程  # 网站优化学着难不难  # 链接对网站优化有什么用  # seo需要懂web 


相关栏目: 【 行业资讯67740 】 【 技术百科0 】 【 网络运营39195


相关推荐: 固态硬盘如何启动  360n6锁屏壁纸怎么设置  服务器系统怎么装  固态硬盘4k如何看  税负是什么意思  什么是unix时间戳  如何以管理员身份打开命令提示符  对应市盈率是30X是什么意思  春运抢票需要什么软件抢  爱玛电动车power模式是什么意思  typescript怎么使用map  如何winpe cmd命令  hive中datediff函数怎么用 Hive中DATEDIFF函数的使用指南  typescript如何标记私有方法  如何查看固态硬盘分区  如何把一个命令后台运行  油电混动车仪表盘上的power是什么意思  春运抢票哪个平台好抢  苹果16最近玩法有哪些  如何用adb命令停用系统软件  65寸电视长宽多少厘米  电焊机power灯亮是什么意思  苹果16改进了哪些  夸克前缀后缀什么意思啊  type-c全能接口是什么意思  typescript变量是什么  如何卸载typescript  电动车eco和power是什么意思  学typescript有什么用  萝卜快跑的收费标准是什么  typescript 如何解决 null  固态硬盘质量如何  手机nfc功能功能是什么意思  j*a数组怎么比较abc  如何设置从固态硬盘启动  cos150度等于多少  春运车站抢票和网上抢票  小屏折叠屏手机有哪些  液位传感器power是什么意思  360n5锁屏壁纸怎么设置  ai如何重复使用上一命令  performance是什么意思  如何管理员打开cmd命令行窗口  摄像机的power chg是什么意思中文  如何用命令查看本机的操作系统  夸克是什么用途  新装固态硬盘如何安装  春运抢票最多能抢几趟车  跨境电商gmv是什么意思?跨境电商GMV:理解其含义、计算方法和影响因素  typescript怎么设置滚动条 

搜索