基于Swin Transformer和形状提取网络流的红外图像小目标检测方法和系统

本发明公开了一种基于Swin Transformer的特征提取网络和形状提取网络流的红外图像小目标检测方法和系统。首先建立基于Swin Transformer的特征提取网络模块和基于泰勒有限差分方程的边缘提取块的形状提取网络模块，上述两个模块并联组成红外图像小目标检测网络模型。将红外图像小目标检测网络模型设置为神经网络，在训练神经网络的过程中实现自动学习红外图像小目标特征以及红外图像小目标特征提取规则；将红外图像输入到小目标检测网络模型，小目标检测网络模型提取小目标的特征并输出小目标在红外图像中的位置以及小目标的形状信息。本发明提取的小目标特征能够充分保持小目标的形状信息，同时准确记录小目标在红外图像中的位置信息。