arXiv:2412.19878cs.CV2024-12被引 52

融合超分辨率与多尺度特征,提升红外小目标检测精度

YOLO-MST: Multiscale deep learning method for infrared small target detection based on super-resolution and YOLO

  • 用超分辨率增强图像,再通过多尺度动态检测头融合特征
  • 在SIRST和IRIS数据集上[email protected]分别达96.4%和99.5%
  • 适合复杂背景中微弱小目标检测,对军事应用有实用价值

随着航空航天技术进步和军事需求增长,低误报率、高精度的红外小目标检测算法成为全球研究重点。传统模型驱动方法在处理噪声、目标尺寸和对比度变化时鲁棒性不足,现有深度学习方法在关键特征提取与融合方面能力有限,在复杂背景或目标特征不明显时难以实现高精度检测。为此,本文提出一种结合图像超分辨率与多尺度观测的深度学习红外小目标检测方法——YOLO-MST。首先对输入红外图像进行超分辨率预处理并多重数据增强;其次,在YOLOv5基础上构建新网络:用自研的MSFA模块替换主干中的SPPF模块,优化颈部结构,并在预测头添加多尺度动态检测头,实现不同尺度特征的动态融合,更好适应复杂场景。该方法在SIRST和IRIS两个公开数据集上的[email protected]分别达到96.4%和99.5%,有效缓解漏检、误报和精度不足问题。

原文摘要 · Abstract (English)

With the advancement of aerospace technology and the increasing demands of military applications, the development of low false-alarm and high-precision infrared small target detection algorithms has emerged as a key focus of research globally. However, the traditional model-driven method is not robust enough when dealing with features such as noise, target size, and contrast. The existing deep-learning methods have limited ability to extract and fuse key features, and it is difficult to achieve high-precision detection in complex backgrounds and when target features are not obvious. To solve these problems, this paper proposes a deep-learning infrared small target detection method that combines image super-resolution technology with multi-scale observation. First, the input infrared images are preprocessed with super-resolution and multiple data enhancements are performed. Secondly, based on the YOLOv5 model, we proposed a new deep-learning network named YOLO-MST. This network includes replacing the SPPF module with the self-designed MSFA module in the backbone, optimizing the neck, and finally adding a multi-scale dynamic detection head to the prediction head. By dynamically fusing features from different scales, the detection head can better adapt to complex scenes. The [email protected] detection rates of this method on two public datasets, SIRST and IRIS, reached 96.4% and 99.5% respectively, more effectively solving the problems of missed detection, false alarms, and low precision.

红外检测小目标多尺度YOLO

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