HairGAN: Spatial-Aware Palette GAN for Hair Color Transfer

Yifan Xie,Yinjia Huang,Ricong Huang,Zhengtao Wu,Mingzhi Mao,Guanbin Li
DOI: https://doi.org/10.1109/icme52920.2022.9859758
2022-01-01
Abstract:Hair color transfer aims to transfer the hair color from a reference image to an original image while maintaining the hair structure of the original image. However, due to the complex hair structure and the misalignment of hair regions between the original image and the reference image, existing methods cannot complete this task well. To address these issues, we propose a hair color transfer GAN (HairGAN). It first utilizes off-the-shelf network to extract the hair region from the original image and transfer it into intermediary hair color. Then, Spatial-Aware Palette Alignment (SAPA) module is introduced to align the hair regions in the original image and the reference image. Furthermore, HairGAN employs cycle consistency reconstruction module to ensure the global consistency between original image and transferred image. We collect a dataset containing 2850 images and a few paired samples for hair color transfer. Extensive experiments show that HairGAN could generate high-quality transferred results.
What problem does this paper attempt to address?