基于小波引导的红外弱小目标检测方法

本发明公开了基于小波引导的红外弱小目标检测方法，具体按照以下步骤实施：步骤1、构建红外弱小目标检测数据集；步骤2、构建编码器，编码器包括卷积层和小波引导Transformer模块；步骤3、构建解码器，解码器包括小波补充上采样模块和全连接层；步骤4、利用数据集训练构建的检测网络，实现对红外弱小目标检测。本发明通过小波引导Transformer模块和小波补充上采样模块，能够有效改善现有技术下目标在上下采样过程中信息丢失问题同时在频域、维度域和空间域三个方向上对目标特征进行提取，得到丰富的细节纹理与语义信息。同时，构建采用该方法的神经网络时通过调整训练参数和损失函数进行网络优化，可以得到更好的红外弱小目标检测效果。