本技术公开一种红外目标检测方法、装置、设备及介质,属于人工智能技术领域。其中红外目标检测方法包括:将红外图像输入至目标检测模型的编码器模块,得到编码特征图,所述目标检测模型为预先训练好的卷积神经网络模型;将所述编码特征图输入至目标检测模型的解码器模块,获取红外图像中待检测目标的掩码;其中,编码器模块包括局部对比增强单元。本发明提出的红外目标检测方法通过计算红外图像中像素点的局部对比注意力权重,引导神经网络在编码过程中强化对潜在红外小目标空间位置的关注,并利用通道注意力强化单元实现对编码特征图的不同通道间信息的有效融合,提高对红外小目标的检测精度。
背景技术
与常用的激光雷达相比,红外探测设备无需主动向外发射探测信号,隐蔽性强,能够在典型作战场景中有效实现远距离目标定位、侦察预警等工作,从而在复杂战场环境中发挥巨大的作用。在图像上对关注的红外弱小目标进行有效的检测一直是该领域的重要且富有挑战的任务。
现有的红外小目标检测方法可大体分为模型驱动(model-driven)的方法和数据驱动(data-driven)的方法两大类。模型驱动的方法主要通过手工构建模型,不需要大量数据的支持,通常遵循特定的假设。模型驱动的方法受复杂背景和噪声的影响较大,且对预设窗口大小、分割阈值等超参数十分敏感,需根据场景特点进行反复的调整尝试,鲁棒性较差。数据驱动的方法主要依赖于深度学习网络,利用监督学习的方法让设计的深度学习网络从数据中自动学习判别特征,在不同场景下均具有相对良好的泛化性。由于红外数据的获取和高质量标注有一定难度,纯数据驱动方法的受数据的制约较大,一定程度上制约了算法的进一步发展。同时,基于CNN和transformer架构的网络通常在参数量上较大,难以将其直接部署于算力较低的边缘端计算设备上。
实现思路