基于颜色形状提示学习的红外小目标识别方法

本申请公开了基于颜色形状提示学习的红外小目标识别方法，包括S1、将红外小目标的图像数据集划分为训练集和测试集；S2、将训练集的图像输入至红外小目标检测网络模型的主干网络，提取含有红外小目标图像的多尺度特征；S3、提取到红外小目标图像的多尺度特征进行多尺度特征增强；S4、通过双重概率对齐DPA融合框架进行异域融合得到多尺度信息特征；S5、将多尺度信息特征送入动态检测头部DynamicHead的进行检测；检测结果由归一化Wasserstein距离NWD和自适应阈值焦点损失ATFL进行评估；S6、获取最终的红外小目标检测结果。本申请具有较好的可靠性和鲁棒性，提高了红外小目标的检测性能。