Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Barbershop: GAN-based Image Compositing using Segmentation Masks

About

Seamlessly blending features from multiple images is extremely challenging because of complex relationships in lighting, geometry, and partial occlusion which cause coupling between different parts of the image. Even though recent work on GANs enables synthesis of realistic hair or faces, it remains difficult to combine them into a single, coherent, and plausible image rather than a disjointed set of image patches. We present a novel solution to image blending, particularly for the problem of hairstyle transfer, based on GAN-inversion. We propose a novel latent space for image blending which is better at preserving detail and encoding spatial information, and propose a new GAN-embedding algorithm which is able to slightly modify images to conform to a common segmentation mask. Our novel representation enables the transfer of the visual properties from multiple reference images including specific details such as moles and wrinkles, and because we do image blending in a latent-space we are able to synthesize images that are coherent. Our approach avoids blending artifacts present in other approaches and finds a globally consistent image. Our results demonstrate a significant improvement over the current state of the art in a user study, with users preferring our blending solution over 95 percent of the time.

Peihao Zhu, Rameen Abdal, John Femiani, Peter Wonka• 2021

Related benchmarks

TaskDatasetResultRank
Color transferCelebA-HQ (test)
Accuracy10.8
7
Self-transfer ReconstructionCelebA-HQ
LPIPS0.11
7
Single Color TransferCelebA-HQ
FID20.54
6
Hairstyle TransferFFHQ (test)
PSNR29.18
4
Hairstyle TransferFFHQ-P (test)
User Preference13.8
4
Hairstyle TransferFFHQ-P 1.0 ([0,15))
User Preference24
4
Hairstyle TransferFFHQ-P 1.0 ([30,45))
User Preference12
4
Hairstyle TransferFFHQ-P 1.0 ([75,90))
User Preference14.7
4
Hairstyle TransferFFHQ All pairs (test)
User Preference16.4
4
Hairstyle TransferFFHQ-S (test)
User Preference17.8
4
Showing 10 of 14 rows

Other info

Follow for update