一种基于结构信息嵌入和注意力机制的数字图像修复方法
Abstract:
本发明公开了一种基于结构信息嵌入和注意力机制的数字图像修复方法,包括如下步骤:首先对公开数据集图像进行尺寸处理、数据增强并采用RTV方法提取结构信息;而后构建基于结构信息嵌入和注意力机制的图像修复网络,利用结构信息及图像特征共享的生成器同时进行结构修复及嵌入指导图像特征重构;利用带光谱归一化的铰链形式结构信息判别器对修复的平滑图像的真伪进行判别;采用联合损失约束模型优化方向;利用训练好的模型权重参数及构建的修复网络可进行对破损图像完成修复任务及效果评估;本发明可以更好地表征结构信息,使得修复后的图片具有更合理的结构及更真实的视觉效果,主客观测试评价结果和对比实验都证明了本发明方法的有效性。
Patent Agency Ranking
0/0