- 专利标题: 一种基于二阶段注意力机制GAN网络图像修复算法
-
申请号: CN202110053892.1申请日: 2021-01-15
-
公开(公告)号: CN112801895A公开(公告)日: 2021-05-14
- 发明人: 王彩玲 , 张晓峰 , 蒋国平
- 申请人: 南京邮电大学
- 申请人地址: 江苏省南京市雨花台区宁双路28号
- 专利权人: 南京邮电大学
- 当前专利权人: 南京邮电大学
- 当前专利权人地址: 江苏省南京市雨花台区宁双路28号
- 代理机构: 南京瑞弘专利商标事务所
- 代理商 陈国强
- 主分类号: G06T5/00
- IPC分类号: G06T5/00 ; G06N3/04 ; G06N3/08
摘要:
本发明公开了一种基于二阶段注意力机制GAN网络图像修复算法,包括以下步骤:搭建基于GPU版本的Pytorch深度学习框架运行环境;准备数据集,每个数据集都有真实的图片和mask;搭建运行图像到图像的深度神经网络,包括生成器和判别器;进行泛化实验。生成器采用编码器-解码器结构,不断训练以生成预测图像;判别器衡量生成图像的真实性,以对抗损失作为损失函数。通过不断迭代联合优化生成器和判别器,以使网络预测性能提高。本发明能够训练实现端到端的自监督图像转换,既能用于学习输入图像到输出图像之间的映射关系,又能学习用于训练映射关系的损失函数。
公开/授权文献
- CN112801895B 一种基于二阶段注意力机制GAN网络图像修复算法 公开/授权日:2022-10-14