一种基于 Transformer 的高光谱点目标检测网络

论文 SpecDETR: A Transformer-based Hyperspectral Point Object Detection Network 提出了一个针对高光谱点目标检测的新型网络 SpecDETR,与传统的基于像素分类的高光谱目标检测方法不同,SpecDETR 从目标检测的角度重新思考了该问题,并专注于目标级预测能力。

图片

亚像素目标与点对象示例。(a) 一张原始遥感图像,地面采样距离为0.28米。(b) 下采样的图像。传统的高光谱目标检测将(b)中组成车辆的8个像素都视为亚像素目标,并预测整个图像中的每个像素是否包含车辆的光谱特征。而我们把这些像素视为一个形状信息有限的统一点对象,并利用对象检测网络来预测该点对象的位置和类别。

1. 从像素级分类到目标级检测的转变:

  • 传统的 hyperspectral target detection (HTD) 方法依赖于预先设定的目标光谱库,对每个像素进行二元分类,判断其是否为目标。
  • SpecDETR 将高光谱点目标检测视为一个目标级检测任务,目标是实现对点目标的实例级检测,并预测其类别。

    图片

    真实世界中的HSI数据,由AVIRIS收集。(a) 飞机。(b) 车辆。(c) 船只。(d) 几种常见物体的光谱辐射曲线。

2. 基于 Transformer 的网络架构:

  • SpecDETR 摒弃了传统目标检测网络中的骨干网络 (backbone),直接利用 Transformer 编码器从光谱标记中提取深层特征。
  • 提出了一种局部和全局协调 (LGC) 注意力模块,该模块在亚像素级别有效地采样局部光谱特征的同时捕获全局光谱特征,确保了背景光谱信息的稳定输入。
  • 设计了一个简洁高效的解码器,并结合了一对多标签分配和 DETR 的端到端优势。

    图片

    SpecDETR概览。为了清晰可视化,我们省略了位置嵌入,只显示最终解码器层的分类预测头。

    图片

    在Avon数据集上,LGC注意力模块的采样点,每个模块有8个注意力头,每个头4个点。(a) 编码器中的自注意力模块。(b) 解码器中的交叉注意力模块。不同颜色表示不同的注意力头。

3.  一对多集合预测:

  • SpecDETR 将点目标检测视为一个一对多集合预测问题,并使用混合一对多标签分配器代替了当前类 DETR 检测器中基于匈牙利匹配的一对一标签分配。
  • 在解码器中,通过移除内容查询之间的交叉注意力计算并使用非学习向量初始化内容查询来简化解码器结构。
  • 重新引入了非极大值抑制 (NMS) 以去除重叠的预测框,增强了 SpecDETR 在点目标检测中的端到端优势。

4. 数据集和实验结果:

在SPOD数据集上的目标模拟设置

图片

图片

SPOD数据集上的伪彩色图像中的物体示例

  • 开发了一个名为 SPOD 的模拟高光谱多类点目标检测数据集,用于评估 SpecDETR 和其他目标检测网络以及 HTD 方法的性能。
  • 在 SPOD 数据集上,SpecDETR 优于最先进的目标检测网络和当前的 HTD 方法,展现出卓越的亚像素目标检测能力。
  • 在一个公开的 HTD 数据集上进行的实验表明,SpecDETR 可以通过基于模拟的数据增强来实现对现实世界中单光谱点目标的检测,无需进行繁琐的手动标注。

5.  红外小目标检测的补充实验:

  • 将 SpecDETR 扩展到红外小目标检测 (IRSTD) 任务,并在三个公共 IRSTD 数据集上评估了 SpecDETR 和其他分割网络的目标级检测性能。
  • 实验结果表明,尽管 SpecDETR 是为点目标检测而设计的,但它仍然具有检测更大、更有形状的目标的能力,并且在目标级检测评估中表现出色。

原创文章,作者:guozi,如若转载,请注明出处:https://www.sudun.com/ask/81368.html

Like (0)
guozi的头像guozi
Previous 2024年5月31日 上午11:36
Next 2024年5月31日 上午11:39

相关推荐

  • 网站被攻击怎么解决,网站被攻击了可以报警吗

    您的网站遭到攻击是任何网站运营商都不愿意面对的问题。然而,无论你是大公司还是个人站长,你都可能会遇到这样的情况。那么,如果您发现您的网站受到攻击,您应该怎么做?在本文中,我们将解释…

    行业资讯 2024年5月7日
    0
  • ip反查的作用及使用方法

    想必大家在使用网络时,都会遇到一些烦恼,比如不知道某个IP地址的具体信息,或者遭遇网络攻击后无法追踪到对方的真实身份。这时候,IP反查就是你的救星。那么什么是IP反查?它有什么作用…

    行业资讯 2024年4月1日
    0
  • 云服务器和本地服务器的优缺点

    你是否曾经听说过云服务器和本地服务器?这两种服务器的区别究竟在哪里?今天,我们将一一为你解开这个谜团。云服务器和本地服务器,它们各自都有着优点和缺点,那么它们到底有什么不同呢?下面…

    行业资讯 2024年3月24日
    0
  • qq代理服务器设置方法(详解)

    你是否曾经遇到过qq无法正常登录的情况?或者想要更快地访问国外网站却被限制?这些问题都可以通过设置qq代理服务器来解决。那么什么是代理服务器?为什么需要设置qq代理服务器?它有哪些…

    行业资讯 2024年4月9日
    0

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注