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

LOHO: Latent Optimization of Hairstyles via Orthogonalization

About

Hairstyle transfer is challenging due to hair structure differences in the source and target hair. Therefore, we propose Latent Optimization of Hairstyles via Orthogonalization (LOHO), an optimization-based approach using GAN inversion to infill missing hair structure details in latent space during hairstyle transfer. Our approach decomposes hair into three attributes: perceptual structure, appearance, and style, and includes tailored losses to model each of these attributes independently. Furthermore, we propose two-stage optimization and gradient orthogonalization to enable disentangled latent space optimization of our hair attributes. Using LOHO for latent space manipulation, users can synthesize novel photorealistic images by manipulating hair attributes either individually or jointly, transferring the desired attributes from reference hairstyles. LOHO achieves a superior FID compared with the current state-of-the-art (SOTA) for hairstyle transfer. Additionally, LOHO preserves the subject's identity comparably well according to PSNR and SSIM when compared to SOTA image embedding pipelines. Code is available at https://github.com/dukebw/LOHO.

Rohit Saha, Brendan Duke, Florian Shkurti, Graham W. Taylor, Parham Aarabi• 2021

Related benchmarks

TaskDatasetResultRank
Hairstyle TransferFFHQ
FID20.72
4
Hairstyle TransferFFHQ-P 1.0 ([45,60))
User Preference12
4
Hairstyle TransferFFHQ-P 1.0 ([75,90))
User Preference10.7
4
Hairstyle TransferFFHQ (test)
PSNR25.76
4
Hairstyle TransferFFHQ All pairs (test)
User Preference8.8
4
Hairstyle TransferFFHQ-P (test)
User Preference9.6
4
Hairstyle TransferFFHQ-S (test)
User Preference8.4
4
Hairstyle TransferFFHQ-P 1.0 ([0,15))
User Preference10.7
4
Hairstyle TransferFFHQ-P [15,30) 1.0
User Preference5.3
4
Hairstyle TransferFFHQ-P 1.0 ([30,45))
User Preference10.7
4
Showing 10 of 13 rows

Other info

Code

Follow for update