一种基于YOLOv4多尺度特征融合的红外小目标检测方法

摘要

本发明公开了一种基于YOLOv4多尺度特征融合的红外小目标检测方法，改善传统红外小目标检测方法在把不同背景下鲁棒性差、特征耦合以及特征易缺失的问题。该发明含有，1、获取红外小目标检测图像进行预处理；2、提取图像的全局特征，对目标区域进行扩张卷积后的特征嵌入全局分支中；3、使用滑动窗口对原始图像进行采样并进行下采样降低原始图像分辨率，得到对抗网络模型输入训练好的生成对抗网络模型；4、将主干网络输出的局部增强的特征、获取的全局特征和上下文特征，通过学习不同的权值相互融合；5、融合后的模块输入空间自注意机制；6、计算损失函数得到最终检测结果。该技术能够适应在不同环境下红外小目标检测的需求。