Nvidia以深度学习自动修复影像,让P图不求人!
报价宝综合消息Nvidia以深度学习自动修复影像,让P图不求人!

图片来源:
Nvidia
Nvidia研究员发表论文,能以深度学习修补破损或是缺少像素的图片,另外也能用于移除部分图片内容,并由人工智能自动填补的方式编辑图片。而这也是第一个能在不规则孔洞中,应用深度学习修复图像的研究。
研究人员提到,过去图像修复的应用,仅能将深度学习技术用在集中于图像中心的矩形孔洞,并且仰赖高成本的后制,但是最新的方法,可以强健的处理任何形状、大小和位置的孔洞,也不需考虑孔洞在图像边界的距离,而且他们的模型经过训练后,可以处理越来越大的孔洞。
这项研究先使用ImageNet、Places2和CelebA-HQ资料集图片来训练其神经网络,该研究团队准备了55,116个随机条纹、任意形状和大小的孔洞遮罩用作训练资料。他们另外还产生了近2.5万个孔洞用来做实验,并且为了增加准确度,还将这些孔洞依据输入图片的大小分为6类。
在训练阶段,透过将不同的孔洞以及缺失的部分叠入完整的图片中,让模型训练学习缺失的像素。在测试阶段则将没使用于训练期间的孔洞引入测试图像的资料集以执行重建的准确性验证。
现存的方法,图片重建的机制会有颜色差异以及模糊的伪像,因为其缺失的像素必然来自于神经网络输入的图像。研究人员强调他们的研究特别避开这个缺陷,他们使用部分卷积层(Partial Convolution Layer),让用来修补的像素,不再只能来自于输入值。这种重新正规化的方法,确保输出值与接受域中缺失的像素值无关。
左图的左下角原本有座桥,经过涂白遮盖影像后,右边由AI自动修复的影像中已看不到桥的踪影。
以AI重建影像:
