中文

PD-GAN:用于图像修复的概率多样性 GAN

计算机视觉与模式识别 2021-05-06 v1

摘要

我们提出了 PD-GAN,一种用于图像修复的概率多样性 GAN(生成对抗网络)。给定带有任意空洞区域的输入图像,PD-GAN 可生成多个内容多样且视觉逼真的修复结果。我们的 PD-GAN 构建于一个基于随机噪声生成图像的原始 GAN 之上。在图像生成过程中,我们通过多尺度注入初始修复图像与空洞区域,对输入随机噪声的深度特征进行由粗到细的调制。我们认为,在空洞填充过程中,靠近空洞边界的像素应更具确定性(即更信赖上下文与初始修复图像以形成自然的修复边界),而位于空洞中心的像素应享有更多自由度(即更依赖于随机噪声以增强多样性)。为此,我们在调制内部提出空间概率多样性归一化(SPDNorm)来建模基于上下文信息生成某一像素的概率。SPDNorm 动态平衡空洞区域内的真实感与多样性,使生成内容在靠近空洞中心处更具多样性,而在靠近边界处更近似邻近图像内容。同时,我们提出一种感知多样性损失以进一步赋予 PD-GAN 生成多样内容的能力。在包括 CelebA-HQ、Places2 和 Paris Street View 在内的基准数据集上的实验表明,PD-GAN 对于多样且视觉逼真的图像修复是有效的。

关键词

引用

@article{arxiv.2105.02201,
  title  = {PD-GAN: Probabilistic Diverse GAN for Image Inpainting},
  author = {Hongyu Liu and Ziyu Wan and Wei Huang and Yibing Song and Xintong Han and Jing Liao},
  journal= {arXiv preprint arXiv:2105.02201},
  year   = {2021}
}

备注

Accepted by CVPR 2021