一种基于空间状态注意力U型网络的红外小目标检测方法及装置

本发明公开了一种基于空间状态注意力U型网络的红外小目标检测方法及装置，属于计算机视觉技术领域，检测方法包括：获取红外小目标图像进行标注处理和增强处理生成图像样本集；构建多级串行的残差网络模块，作为编码器对样本图像进行特征提取；构建多级串行的空间状态注意力模块，作为解码器对编码器输出的特征图像进行图像重构；将编码器输出的特征图像和解码器输出的重构图像进行跨通道融合并与样本图像对应的标签图像计算损失；基于损失通过反向传播和梯度下降算法对编码器和解码器进行网络参数训练更新；通过训练好的编码器和解码器对待测红外小目标图像进行检测，对检测结果进行灰度处理并输出。本发明能够实现准确高效的红外小目标检测。