基于扩散模型的二阶段细化图像修复模型

来源：电子测量技术杂志2025年第17期北京时间：

作者：张绪，胡峻峰

单位：东北林业大学计算机与控制工程学院 哈尔滨 150040

摘要：近年来，图像生成领域的技术取得了显著进展，但图像修复任务中修复区域与未修改区域之间的一致性仍是一个普遍存在的挑战。本文旨在提出一种基于扩散模型的两阶段图像修复模型（Diff-2sIR），以提升修复区域与未修复区域的一致性，进一步提高图像修复质量。本文以扩散模型理论为基础，设计了一种两阶段修复框架。通过改进U-Net架构和扩散模型采样算法，对初步修复结果进行二次细化修复，缓解了修复区域与未修复区域之间的不一致性问题。在CelebA-HQ数据集的人脸修复任务中，Diff-2sIR模型取得了最优FID分数（2.92），显著提升了修复质量。实验结果表明，该模型在指导模块修复结果的基础上进一步细化修复效果，展现了卓越的性能。本文提出的Diff-2sIR模型有效解决了修复区域与未修复区域之间的不一致性问题，为图像修复任务提供了一种新的解决方案，具有重要的理论意义和应用价值。

关键词：图像修复;图像生成;扩散模型;二阶段模型

基金资助：国家自然科学基金（32202147）、中央高校基本科研专项资金项目（2572019BF09）资助