一种基于扩散模型的图像生成方法

本公开实施例公开了一种基于扩散模型的图像生成方法，获取第一干净图像输入扩散模型的扩散过程，得到噪声图像，扩散模型包括Wavelet块并基于人脸五官特征构建扩散模型的损失函数，将噪声图像输入扩散模型的重建过程，对扩散模型进行训练，得到目标扩散模型，获取随机高斯噪声图像，将随机高斯噪声图像输入目标扩散模型的重建过程，得到生成的人脸图像。在扩散模型中引入Wavelet块，可以使用Wavelet块提取第一特征图的第一频率特征，也就是提取噪声图像的高频和低频信息，扩散模型根据高频和低频信息分离高斯噪声，能够有效分离高斯噪声，提高了生成的人脸图像的质量，且基于人脸五官特征构建损失函数，提升了生成的人脸图像的五官细节部分的质量。