基于扩散模型的梯度引导图像修复算法研究
DOI:
CSTR:
作者:
作者单位:

1.西安石油大学电子工程学院 西安 710300; 2.西安石油大学化学化工学院 西安 710300

作者简介:

通讯作者:

中图分类号:

TP391.41;TN0

基金项目:

国家自然科学基金(22373075)项目资助


Gradient-guided image inpainting algorithm based on diffusion models
Author:
Affiliation:

1.School of Electronic Engineering, Xi′an Shiyou University,Xi′an 710300, China; 2.School of Chemistry and Chemical Engineering, Xi′an Shiyou University,Xi′an 710300, China

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    去噪扩散概率模型(DDPM)展示了强大的图像生成能力,并成功应用于图像修复。近几年流行的方法通过引入结构先验来修复图像,尽管取得了良好的表现,但在结构还原的完整性与纹理细节的自然性方面仍存在不足,常出现断裂或不连续的修复区域。针对此不足之处,提出了一种基于扩散模型的梯度引导特征重建算法。首先对破损图像的梯度图进行修复,初步重建图像的纹理和结构;随后将修复后的梯度图作为生成引导,辅助原图进行修复;最后在噪声预测网络中引入高效通道注意力(ECA)模块,优化特征交互,进一步提升修复图像的一致性。实验结果表明,与主流算法相比,在CelebA-HQ数据集上,所提方法在峰值信噪比(PSNR)和结构相似性(SSIM)指标上分别提高3.19% 和2.74%,感知相似度(LPIPS)指标下降8.82%;在Places2数据集上,PSNR和SSIM分别提高0.42% 和2.81%,LPIPS下降2.75%,证明所提方法在图像修复任务中具备更强的结构还原与细节保持能力。

    Abstract:

    Denoising diffusion probabilistic models(DDPM)have demonstrated powerful image generation capabilities and have been successfully applied to image inpainting. While many recent approaches introduce structural priors to assist the inpainting process and have achieved promising results, they still suffer from limitations in structural integrity and the naturalness of texture details, often resulting in fractured or discontinuous regions. To address these issues, this paper proposes a gradient-guided feature reconstruction algorithm based on diffusion models. Specifically, the gradient map of the corrupted image is first restored to preliminarily reconstruct its structure and texture. The restored gradient map is then used as generation guidance to assist the inpainting of the original image. Furthermore, an efficient channel attention (ECA) module is integrated into the noise prediction network to enhance feature interaction and improve the consistency of the reconstructed images. Experimental results show that compared with state-of-the-art methods, the proposed approach achieves improvements of 3.19% in PSNR and 2.74% in SSIM, and a reduction of 8.82% in LPIPS on the CelebA-HQ dataset. On the Places2 dataset, it achieves gains of 0.42% in PSNR and 2.81% in SSIM, with an 2.75% decrease in LPIPS, demonstrating superior capability in structural restoration and detail preservation for image inpainting tasks.

    参考文献
    相似文献
    引证文献
引用本文

张荣铠,景明利,焦龙.基于扩散模型的梯度引导图像修复算法研究[J].电子测量技术,2026,49(10):197-205

复制
分享
相关视频

文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2026-08-25
  • 出版日期:
文章二维码

重要通知公告

①《电子测量技术》期刊收款账户变更公告