一种基于对抗性边缘学习重建的图像修复方法

文档序号:36027421发布日期:2023-11-17 15:22阅读:来源:国知局

技术特征:

1.一种基于对抗性边缘学习重建的图像修复方法,其特征在于:包括,

2.如权利要求1所述的一种基于对抗性边缘学习重建的图像修复方法,其特征在于:所述预处理包括,通过数据集的选取与处理,进行不规则mask和人脸数据集的载入,并进行图像的预处理与划分。

3.如权利要求2所述的一种基于对抗性边缘学习重建的图像修复方法,其特征在于:所述图像的预处理包括对于图像的边缘提取,使用canny边缘检测获取图像的边缘,具体如下:

4.如权利要求3所述的一种基于对抗性边缘学习重建的图像修复方法,其特征在于:所述搭建模型包括在边缘生成器中使用掩膜灰度图像作为输入,其边缘图以及掩膜m作为前提条件:

5.如权利要求4所述的一种基于对抗性边缘学习重建的图像修复方法,其特征在于:所述搭建模型还包括,所使用的网络利用不完整的彩色图像作为输入,并使用复合边缘图ccomp作为条件,复合边缘图ccomp是通过将真实边缘图在背景区域与上一阶段生成的边缘图在缺失区域进行组合得到的:ccomp=cgt⊙(1-m)+cpred⊙m,网络返回一张修复后的彩色图像ipred:

6.如权利要求5所述的一种基于对抗性边缘学习重建的图像修复方法,其特征在于:所述损失包括重建损失、对抗损失、感知损失以及风格损失;

7.如权利要求6所述的一种基于对抗性边缘学习重建的图像修复方法,其特征在于:所述训练模型包括采用反向传播算法更新优化网络参数权重,直至模型损失趋于收敛。

8.如权利要求7所述的一种基于对抗性边缘学习重建的图像修复方法,其特征在于:所述训练模型还包括将训练好的模型导入并使用opencv工具做成人脸涂抹修复软件,最后使用打包工具打包成exe文件。

9.一种计算机设备,包括存储器和处理器,所述存储器存储有计算机程序,其特征在于,所述处理器执行所述计算机程序时实现权利要求1至8中任一项所述的方法的步骤。

10.一种计算机可读存储介质,其上存储有计算机程序,其特征在于,所述计算机程序被处理器执行时实现权利要求1至8中任一项所述的方法的步骤。


技术总结
本发明公开了一种基于对抗性边缘学习重建的图像修复方法包括,准备数据集并且对数据集进行训练前的预处理;搭建模型,采用经典的生成式对抗网络,生成器和判别器进行对抗,并且将生成器的输出结果与输入图像进行残差相加,分别训练边缘模型,修补模型;设置模型训练约束;进行模型训练。本方法对于传统修复的图像中有明显的缺失与不自然痕迹有了改进,使得痕迹更加不明显,效果更令人满意。使用opencv做了简易的前端系统,并且使用python工具进行打包,使得在无环境配置,无编程经验的电脑上也能运行,解决传统的修复方法在细节的修复中有很大比例的缺失,修复痕迹较为明显,且如果涂抹的地方太小,效果更差,并且方法环境配置较为繁琐的问题。

技术研发人员:胡聪,刘泰麟,罗卓毅,王悦,刘浩迪,李佳乐
受保护的技术使用者:江南大学
技术研发日:
技术公布日:2024/1/16
当前第2页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1