基于改进的密集嵌套U-Net网络的红外小目标检测算法
Sensors (Basel, Switzerland)
|February 13, 2025
概括
本研究介绍了IDNA-UNet,这是一个改进的深度学习模型,用于检测小型红外目标. 它增强了特征融合,并使用尺度敏感损失来显著提高检测准确性并减少错误报警.
科学领域:
- 计算机视觉 计算机视觉
- 机器学习 机器学习
- 红外成像技术 红外成像技术
背景情况:
- 红外小型目标检测对于早期预警,医疗诊断和反无人机系统至关重要.
- 卷积神经网络 (CNN) 方法在小型红外目标上遇到困难,原因是下采样导致的特征损失.
研究的目的:
- 解决红外小目标检测现有方法的局限性.
- 提出一个改进的基于U-Net的模型,用于增强小目标检测.
主要方法:
- 开发了一种基于U-Net (IDNA-UNet) 的改进密集嵌套和注意力红外小目标检测方法.
- 引入了密集嵌套交互模块 (DNIM),用于逐级的特征融合和小目标特征的保留.
- 实现了一个自下而上的特征金字塔融合模块,用于整合低级和高级特征.
- 使用尺度和位置敏感 (SLS) 损失函数来准确定位和尺度差异化.
主要成果:
- 通过重复的融合和增强,IDNA-UNet有效地结合和利用小目标的上下文信息.
- 拟议的方法在交叉与联合 (IoU),检测概率 (Pd) 和错误报警率 (Fa) 上显示出明显的优势.
结论:
- 与现有的方法相比,IDNA-UNet在红外小型目标检测方面提供了更高的性能.
- 该模型的架构和损失函数有助于在具有挑战性的检测场景中提高准确性和可靠性.


