一种基于数据和特征的红外小目标检测优化方法及系统

本发明涉及一种基于数据和特征的红外小目标检测优化方法及系统，属于计算机视觉技术领域。首先预处理红外小目标数据集中的训练数据，并对训练数据进行分类，随机裁剪扩充数据量。将分类好的数据和其对应的标签输入扩散模型进行训练。然后，加载训练好的扩散模型参数，输入随机噪声、随机类别和随机标签生成红外数据，融入原始训练数据，得到新的合成训练数据集。在神经网络中，应用基于记忆的检测头，使用合成数据集进行训练。最后加载训练好的网络参数，输入需要检测的红外图像，得到检测结果。本发明引入了扩散模型来生成红外数据，增加了训练数据的多样性，提高了检测精度和稳定性。