基于条件扩散模型的红外小目标检测方法

本发明公开了一种基于条件扩散模型的红外小目标检测方法。具体而言，所述条件扩散模型包括条件感知编码模块、条件引导的特征融合编码模块和特征解码模块；该方法将红外图像经过条件引导的条件感知编码模块进行多尺度特征提取与融合生成条件特征，条件特征与时间步t的噪声掩码进行特征融合，引导模型定向优化目标掩码。在条件扩散模型的训练过程中，采用基于时间步t的均方误差损失函数对生成掩码的准确性进行监督，并通过Adam优化器动态调整模型参数以最小化噪声预测误差。目标检测阶段，通过逆向扩散过程的马尔科夫链迭代，从高斯噪声的初始状态逐步去噪，最终生成高精度目标掩码。