一种基于扩散模型的建筑纹理图像修复方法

专利主权项内容

1.一种基于扩散模型的建筑纹理图像修复方法，其特征在于，包括以下步骤：S1、以原始建筑纹理图像作为扩散模型在前向过程的输入数据，得到标准高斯噪声数据；S2、以添加高斯噪声的建筑纹理图像和标准高斯噪声数据作为扩散模型在逆向过程的输入数据，在扩散模型的逆向过程中添加编码神经网络和预测神经网络，利用编码神经网络对添加高斯噪声的建筑纹理图像进行编码，并利用预测神经网络根据编码图像和数据分布预测新的高斯噪声数据和数据分布，进行扩散模型训练；S3、以待修复的建筑纹理图像和标准高斯噪声数据作为扩散模型在添加编码神经网络和预测神经网络的逆向过程的输入数据，得到修复后的建筑纹理图像。