arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2207.09367cs.CV
收藏

用于改进重构和可编辑性的StyleGAN编码器的循环编码

Cycle Encoding of a StyleGAN Encoder for Improved Reconstruction and Editability

Sun Yat-sen University(中山大学);Xiamen University(厦门大学);City University of Hong Kong(香港城市大学);Guangdong University of Technology(广东工业大学);Hong Kong Polytechnic University(香港理工大学)

Xudong Mao, Liujuan Cao, Aurele T. Gnanha, Zhenguo Yang, Qing Li, Rongrong Ji

摘要:

GAN反演旨在将输入图像反演到预先训练的GAN的潜在空间中。尽管最近在GAN反演中取得了进展,但仍然存在减轻失真和可编辑性之间的权衡的挑战,即准确地重建输入图像并以小的视觉质量下降编辑反演图像。最近提出的枢轴调整模型,通过使用两步的方法,首先将输入图像反转成一个潜在的代码,称为枢轴代码,然后改变生成器,使输入图像可以准确地映射到枢轴代码,使重建和可编辑性取得了重大进展。在这里,我们表明,重建和可编辑性可以通过适当的设计枢轴代码。我们提出了一个简单而有效的方法,命名为循环编码,高质量的枢轴代码。我们的方法的核心思想是逐步训练的编码器在不同的空间根据一个循环计划:W->W+->W。这种训练方法保留了W和W+空间的属性,即W的高可编辑性和W+的低失真。为了进一步减少失真,我们还建议使用基于优化的方法来改进枢轴代码,其中引入正则化项以减少可编辑性的退化。定性和定量比较几个国家的最先进的方法证明了我们的方法的优越性。

英文摘要:

GAN inversion aims to invert an input image into the latent space of a pre-trained GAN. Despite the recent advances in GAN inversion, there remain challenges to mitigate the tradeoff between distortion and editability, i.e. reconstructing the input image accurately and editing the inverted image with a small visual quality drop. The recently proposed pivotal tuning model makes significant progress towards reconstruction and editability, by using a two-step approach that first inverts the input image into a latent code, called pivot code, and then alters the generator so that the input image can be accurately mapped into the pivot code. Here, we show that both reconstruction and editability can be improved by a proper design of the pivot code. We present a simple yet effective method, named cycle encoding, for a high-quality pivot code. The key idea of our method is to progressively train an encoder in varying spaces according to a cycle scheme: W->W+->W. This training methodology preserves the properties of both W and W+ spaces, i.e. high editability of W and low distortion of W+. To further decrease the distortion, we also propose to refine the pivot code with an optimization-based method, where a regularization term is introduced to reduce the degradation in editability. Qualitative and quantitative comparisons to several state-of-the-art methods demonstrate the superiority of our approach.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑