基于多尺度注意力特征叠加的红外小目标检测模型建立方法及检测方法

    公开(公告)号:CN118968012A

    公开(公告)日:2024-11-15

    申请号:CN202410877321.3

    申请日:2024-07-02

    摘要: 本发明公开了基于多尺度注意力特征叠加的红外小目标检测模型建立方法及检测方法,属于红外小目标检测领域,模型建立方法包括:获取包含红外小目标的红外图像集,对各图像中小目标所在的目标框,以及目标像素和背景像素进行标注;利用标注后的红外图像集对初始检测网络进行训练,得到红外小目标检测模型;初始检测网络中,多尺度区域候选网络用于提取输入图像不同尺度的特征,提取各尺度特征的空间注意力后,基于空间注意力将不同尺度的特征融合到一起,并由此筛选出包含红外小目标的目标框;语义提取器获取目标框的语义特征后,由分割编码器识别出其中各像素属于小目标的概率,得到红外小目标检测结果。本发明能够提高红外小目标的检测质量。