带式输送机纵向撕裂红外图像数据生成方法

本发明提供了一种带式输送机纵向撕裂红外图像数据生成方法，包括以下步骤：步骤S1：在健康红外图像上随机选择纵向撕裂发生区域；步骤S2：从原始带式输送机的纵向撕裂特征图像中截取不同尺寸的纵向撕裂特征，采用纵向撕裂特征对纵向撕裂发生区域进行填充，得到合成图像；步骤S3：对合成图像的特征边缘进行裁剪，采用通过原始带式输送机纵向撕裂红外图像进行训练后的去噪扩散概率模型进行修复；步骤S4：将修复后的合成图像和原始带式输送机纵向撕裂红外图像进行混合，对去噪扩散概率模型进行二次训练后进行采样进行数据增强以生成图像数据。实现少样本情况下对纵向撕裂数据集的高质量、多类型扩充。