Unsupervised Image-to-Image Translation via Pre-Trained StyleGAN2 Network

被引:46
作者
Huang, Jialu [1 ]
Liao, Jing [1 ]
Kwong, Sam [1 ,2 ]
机构
[1] City Univ Hong Kong, Dept Comp Sci, Kowloon, Hong Kong, Peoples R China
[2] City Univ Hong Kong Shenzhen Res Inst, Kowloon, Hong Kong, Peoples R China
基金
中国国家自然科学基金;
关键词
Training; Gallium nitride; Data models; Training data; Image reconstruction; Task analysis; Superresolution; GAN; machine learning; neural network;
D O I
10.1109/TMM.2021.3065230
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Image-to-Image (I2I) translation is an emerging topic in academia, and it also has been applied in real-world industry for tasks like image synthesis, super-resolution, and colorization. Traditional I2I translation methods usually train data in two or more domains together. This requires lots of computation resources. The results are of lower quality, and contain more artifacts. The training process could be unstable when the data in different domains are not balanced, and modal collapse is more likely to happen. In this paper, we propose a new I2I translation method that generates a new model in the target domain via a series of model transformations on a pre-trained StyleGAN2 model in the source domain. After that, we develop an inversion method to achieve the conversion between an image and its latent vector. By feeding the latent vector into the generated model, we can perform I2I translation between the source domain and target domain. Both qualitative and quantitative evaluations were conducted to verify that the proposed method can achieve better performance in terms of image quality, diversity and semantic similarity to the input and reference images compared to state-of-the-art works.
引用
收藏
页码:1435 / 1448
页数:14
相关论文
共 60 条
[1]   Image2StyleGAN++: How to Edit the Embedded Images? [J].
Abdal, Rameen ;
Qin, Yipeng ;
Wonka, Peter .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2020), 2020, :8293-8302
[2]   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
[3]  
Adler Doron., TOON YOURS
[4]  
[Anonymous], 2019, DANBOORU2018 ALARGE
[5]   Transfer learning on fused multiparametric MR images for classifying histopathological subtypes of rhabdomyosarcoma [J].
Banerjee, Imon ;
Crawley, Alexis ;
Bhethanabotla, Mythili ;
Daldrup-Link, Heike E. ;
Rubin, Daniel L. .
COMPUTERIZED MEDICAL IMAGING AND GRAPHICS, 2018, 65 :167-175
[6]   FlyMap: Interacting with Maps Projected from a Drone [J].
Brock, Anke M. ;
Chatain, Julia ;
Park, Michelle ;
Fang, Tommy ;
Hachet, Martin ;
Landay, James A. ;
Cauchard, Jessica R. .
PROCEEDINGS PERVASIVE DISPLAYS 2018: THE 7TH ACM INTERNATIONAL SYMPOSIUM ON PERVASIVE DISPLAYS, 2018,
[7]   Quality-Aware Unpaired Image-to-Image Translation [J].
Chen, Lei ;
Wu, Le ;
Hu, Zhenzhen ;
Wang, Meng .
IEEE TRANSACTIONS ON MULTIMEDIA, 2019, 21 (10) :2664-2674
[8]   Inverting the Generator of a Generative Adversarial Network [J].
Creswell, Antonia ;
Bharath, Anil Anthony .
IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2019, 30 (07) :1967-1974
[9]   SPA-GAN: Spatial Attention GAN for Image-to-Image Translation [J].
Emami, Hajar ;
Aliabadi, Majid Moradi ;
Dong, Ming ;
Chinnam, Ratna Babu .
IEEE TRANSACTIONS ON MULTIMEDIA, 2021, 23 :391-401
[10]   Transfer Learning with Partial Observability Applied to Cervical Cancer Screening [J].
Fernandes, Kelwin ;
Cardoso, Jaime S. ;
Fernandes, Jessica .
PATTERN RECOGNITION AND IMAGE ANALYSIS (IBPRIA 2017), 2017, 10255 :243-250