arXiv:2505.19148cs.CV2025-05ICCV被引 6

首个专用于红外小目标解混的深度学习模型,实现亚像素级精准定位。

DISTA-Net: Dynamic Closely-Spaced Infrared Small Target Unmixing

  • 动态调整卷积权重与阈值,实时优化信号分离过程。
  • 在10万样本数据集上达到领先亚像素检测精度,优于现有方法。
  • 开源数据集、评估指标与工具链,推动领域发展。

密集集群中的紧密相邻小目标在红外成像中难以分辨,因信号重叠导致数量、亚像素位置及辐射强度难以精确确定。尽管深度学习已推进红外小目标检测,但对紧密相邻目标的解混尚未探索,主要受限于特征叠加复杂性及缺乏开源基础设施。本文提出动态迭代收缩阈值网络(DISTA-Net),将传统稀疏重建重构为动态框架,实时自适应生成卷积权重与阈值参数以优化重建过程。据我们所知,DISTA-Net是首个专为紧密相邻红外小目标解混设计的深度学习模型,实现了优越的亚像素检测精度。此外,我们建立了首个开源生态体系,包含:(1) 公开基准数据集CSIST-100K;(2) 针对亚像素检测定制的评价指标CSO-mAP;(3) 开源工具包GrokCSO,集成DISTA-Net及其他模型。代码与数据集见https://github.com/GrokCV/GrokCSO。

原文摘要 · Abstract (English)

Resolving closely-spaced small targets in dense clusters presents a significant challenge in infrared imaging, as the overlapping signals hinder precise determination of their quantity, sub-pixel positions, and radiation intensities. While deep learning has advanced the field of infrared small target detection, its application to closely-spaced infrared small targets has not yet been explored. This gap exists primarily due to the complexity of separating superimposed characteristics and the lack of an open-source infrastructure. In this work, we propose the Dynamic Iterative Shrinkage Thresholding Network (DISTA-Net), which reconceptualizes traditional sparse reconstruction within a dynamic framework. DISTA-Net adaptively generates convolution weights and thresholding parameters to tailor the reconstruction process in real time. To the best of our knowledge, DISTA-Net is the first deep learning model designed specifically for the unmixing of closely-spaced infrared small targets, achieving superior sub-pixel detection accuracy. Moreover, we have established the first open-source ecosystem to foster further research in this field. This ecosystem comprises three key components: (1) CSIST-100K, a publicly available benchmark dataset; (2) CSO-mAP, a custom evaluation metric for sub-pixel detection; and (3) GrokCSO, an open-source toolkit featuring DISTA-Net and other models. Our code and dataset are available at https://github.com/GrokCV/GrokCSO.

红外成像小目标检测解混亚像素

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