一种基于Swin-Transformer和多尺度特征融合的红外弱小目标检测方法

    公开(公告)号:CN116188944A

    公开(公告)日:2023-05-30

    申请号:CN202310205449.0

    申请日:2023-03-06

    Abstract: 本发明涉及红外弱小目标检测技术领域,具体涉及一种基于Swin‑Transformer和多尺度特征融合的红外弱小目标检测方法,包括:在Unet网络中引入Swin‑Transformer模块来替代原有的卷积层进行特征提取,以构成目标检测模型;将待检测的红外图像输入经过训练的目标检测模型中:首先通过多个Swin‑Transformer模块逐层提取红外图像的特征信息,生成多个尺度的特征图;然后通过多个跨层特征融合模块从最高尺度的特征图开始,依次融合各个尺度的特征图,生成对应的多层融合特征图;最后将多层融合特征图输入分类器中进行归一化处理,并输出对应的目标预测结果。本发明能够保证检测模型在复杂背景、低信噪比等场景下的检测性能,并且能够降低检测模型随着网络加深而丢失红外弱小目标空间细节的风险。

Patent Agency Ranking