arXiv:2412.16986cs.CV2024-12AAAI被引 275

提出新型螺旋卷积与动态损失,提升红外小目标检测精度

Pinwheel-shaped Convolution and Scale-based Dynamic Loss for Infrared Small Target Detection

  • 用螺旋形卷积替代标准卷积,更好捕捉小目标像素分布特征
  • 在IRSTD-1K和新数据集SIRST-UAVB上实现显著性能提升
  • 适合需要高精度红外小目标检测的科研与工程应用

近年来基于卷积神经网络的红外小目标检测方法取得了优异性能。然而,这些方法通常采用标准卷积,未考虑红外小目标像素分布的空间特性。为此,本文提出一种新型螺旋形卷积(PConv),用于骨干网络低层替代标准卷积。PConv更契合小目标像素的高斯空间分布,增强特征提取能力,显著扩大感受野,且参数增加极少。此外,现有损失函数虽结合尺度与位置损失,但未充分考虑不同目标尺度下损失敏感度差异,限制了对微弱小目标的检测效果。为此,提出基于尺度的动态损失(SD Loss),根据目标大小动态调整尺度与位置损失的影响,提升网络对多尺度目标的检测能力。构建了新基准SIRST-UAVB,是目前最大最挑战的单帧红外小目标检测数据集。将PConv与SD Loss集成至最新检测算法,在IRSTD-1K和SIRST-UAVB上均取得显著提升,验证了方法的有效性与泛化能力。

原文摘要 · Abstract (English)

These recent years have witnessed that convolutional neural network (CNN)-based methods for detecting infrared small targets have achieved outstanding performance. However, these methods typically employ standard convolutions, neglecting to consider the spatial characteristics of the pixel distribution of infrared small targets. Therefore, we propose a novel pinwheel-shaped convolution (PConv) as a replacement for standard convolutions in the lower layers of the backbone network. PConv better aligns with the pixel Gaussian spatial distribution of dim small targets, enhances feature extraction, significantly increases the receptive field, and introduces only a minimal increase in parameters. Additionally, while recent loss functions combine scale and location losses, they do not adequately account for the varying sensitivity of these losses across different target scales, limiting detection performance on dim-small targets. To overcome this, we propose a scale-based dynamic (SD) Loss that dynamically adjusts the influence of scale and location losses based on target size, improving the network's ability to detect targets of varying scales. We construct a new benchmark, SIRST-UAVB, which is the largest and most challenging dataset to date for real-shot single-frame infrared small target detection. Lastly, by integrating PConv and SD Loss into the latest small target detection algorithms, we achieved significant performance improvements on IRSTD-1K and our SIRST-UAVB dataset, validating the effectiveness and generalizability of our approach. Code -- https://github.com/JN-Yang/PConv-SDloss-Data

红外检测小目标卷积设计动态损失

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。