HyperInverter: Improving StyleGAN Inversion via Hypernetwork

被引:77
作者
Dinh, Tan M. [1 ]
Anh Tuan Tran [1 ]
Rang Nguyen [1 ]
Binh-Son Hua [1 ]
机构
[1] VinAI Res, Hanoi, Vietnam
来源
2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR) | 2022年
关键词
D O I
10.1109/CVPR52688.2022.01110
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Real-world image manipulation has achieved fantastic progress in recent years as a result of the exploration and utilization of GAN latent spaces. GAN inversion is the first step in this pipeline, which aims to map the real image to the latent code faithfully. Unfortunately, the majority of existing GAN inversion methods fail to meet at least one of the three requirements listed below: high reconstruction quality, editability, and fast inference. We present a novel two-phase strategy in this research that fits all requirements at the same time. In the first phase, we train an encoder to map the input image to StyleGAN2 W-space, which was proven to have excellent editability but lower reconstruction quality. In the second phase, we supplement the reconstruction ability in the initial phase by leveraging a series of hypernetworks to recover the missing information during inversion. These two steps complement each other to yield high reconstruction quality thanks to the hypernetwork branch and excellent editability due to the inversion done in the W-space. Our method is entirely encoder-based, resulting in extremely fast inference. Extensive experiments on two challenging datasets demonstrate the superiority of our method.(1)
引用
收藏
页码:11379 / 11388
页数:10
相关论文
共 59 条
[1]   Image2StyleGAN: How to Embed Images Into the StyleGAN Latent Space? [J].
Abdal, Rameen ;
Qin, Yipeng ;
Wonka, Peter .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, :4431-4440
[2]  
Agarap A.F., 2018, CoRR abs/1803.08375
[3]  
Alaluf Yuval, 2021, HyperStyle: styleGAN inversion with HyperNetworks for real image editing
[4]  
Alaluf Yuval, 2021, ICCV
[5]  
[Anonymous], 2016, Hypernetworks
[6]  
[Anonymous], 2020, CVPR, DOI DOI 10.1109/CVPR42600.2020.00640
[7]   Understanding the role of individual units in a deep neural network [J].
Bau, David ;
Zhu, Jun-Yan ;
Strobelt, Hendrik ;
Lapedriza, Agata ;
Zhou, Bolei ;
Torralba, Antonio .
PROCEEDINGS OF THE NATIONAL ACADEMY OF SCIENCES OF THE UNITED STATES OF AMERICA, 2020, 117 (48) :30071-30078
[8]   Seeing What a GAN Cannot Generate [J].
Bau, David ;
Zhu, Jun-Yan ;
Wulff, Jonas ;
Peebles, William ;
Strobelt, Hendrik ;
Zhou, Bolei ;
Torralba, Antonio .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, :4501-4510
[9]  
Bau David, 2019, TOG
[10]  
Binkowski M., 2018, INT C LEARNING REPRE