一种基于目标框生成红外图像的方法、扩散模型及系统

本发明提供了一种基于目标框生成红外图像的方法、扩散模型及系统，该方法包括：获取物体的目标框；得到与目标框类别对应的类别编码特征；得到与目标框位置对应的位置编码特征；得到目标框编码特征；对红外图像的隐空间特征进行前向加噪，得到加噪完成的特征；基于目标框编码特征获取扩散模型的去噪子模型的损失函数；基于扩散模型的去噪子模型的损失函数、加噪完成的特征对扩散模型的去噪子模型进行去噪训练，得到训练好的扩散模型；训练好的扩散模型基于待测试物体的目标框对应的目标框编码特征生成对应的红外图像。本发明能够生成趋近于真实的红外图像，并能在给定的目标框范围内准确生成对应物体。