首页|基于去噪扩散概率模型的人脸图像修复模型

基于去噪扩散概率模型的人脸图像修复模型

Face Inpainting Model Based on Denoising Diffusion Probability Models

扫码查看
针对使用主流人脸图像修复模型在修复图像后,出现图像质量欠佳、修复边缘模糊,且模型复杂、训练困难的问题,提出了一种基于去噪扩散概率模型的人脸图像修复模型.通过使用Guided-diffusion中的U-Net网络结构,并在网络中引入快速傅里叶卷积来改进去噪扩散概率模型,最后在CelebA-HQ高清人脸图像数据集上进行模型的训练与结果评估.实验结果表明,改进后的去噪扩散概率模型在修复随机掩码的人脸图像时,修复结果与原图的PSNR(峰值信噪比)可以达到25.01,SSIM(结构相似性)可以达到0.886,优于改进前的去噪扩散概率模型与现有的基于生成对抗网络的人脸图像修复模型.
A face inpainting model based on the denoising diffusion probability model is proposed aiming at the problems of poor image quality,blurred repair edges,complex model,and difficult training of the mainstream face inpainting model after image inpainting.By improving the denoising diffusion probability model,the U-Net network structure in Guided-diffusion is adopted.The fast Fourier convolution is introduced into the network,and then the model is trained and tested on the CelebA-HQ high-definition face image dataset.The experimental results show that the improved denoising diffusion probability model can achieve a PSNR of 25.01 and a SSIM of 0.886 compared to the original image,when inpainting face images with random mask,both of which are better than the model before improvement and the existing face image inpainting model based on generative adversarial networks.

deep learningface inpaintingdenoising diffusion probability modelsfast Fourier convolutionU-Net network

刘纪红、黄熙雄

展开 >

东北大学 信息科学与工程学院,辽宁 沈阳 110819

深度学习 人脸图像修复 去噪扩散概率模型 快速傅里叶卷积 U-Net网络

2024

东北大学学报(自然科学版)
东北大学

东北大学学报(自然科学版)

CSTPCD北大核心
影响因子:0.507
ISSN:1005-3026
年,卷(期):2024.45(9)