atlas

Research map / Image generation / Editing and style transfer

HairFastGAN: Realistic and Robust Hair Transfer with a Fast Encoder-Based Approach

NeurIPS 2024 · Image generation › Editing and style transfer · approach: Arbitrary neural style transfer · 4 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes HairFast, an encoder-only hairstyle and color transfer pipeline operating in StyleGAN's FS space with pose alignment, mixing, and refinement modules, achieving near real-time inference.

Problem

Hairstyle transfer from a reference photo to an input face is slow when optimization-based (Barbershop, StyleYourHair, HairCLIPv2) and low quality when encoder-based (HairCLIP in W+), and both struggle with large pose differences between source and reference.

Approach

Decomposes hair transfer into pose alignment (Rotate Encoder in W+ plus shape encoder/adaptor for masks), shape alignment (F-space assembly with SEAN inpainting and E4E regularization), color alignment (S-space modulation encoder with CLIP embeddings and a CLIP cosine loss), and a refinement FS encoder at 64x64 for detail recovery, all using feed-forward encoders with no test-time optimization.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use HairFast instead of HairCLIPv2Hair transfer · CelebA-HQ hair transfer (full)FID: 14.28 -> 13.12Helped
Use HairFast instead of BarbershopHair transfer · CelebA-HQ hair transfer (full)FID_CLIP: 7.07 -> 5.12Helped
Use HairFast instead of HairCLIPHair transfer · Hair transfer runtime (V100)time (s): 0.36 -> 0.78Hurt
Add Color alignment (ablation)HairFast · CelebA-HQ hair transferFID: 26.88 -> 16.23Helped
Add FS and W+ F-space mixing (ablation)HairFast · CelebA-HQ hair transferFID: 16.57 -> 16.23No clear effect
Add Rotate Encoder (ablation)HairFast · CelebA-HQ hair transferFID: 16.87 -> 16.23Helped
Add Shape Adaptor (ablation)HairFast · CelebA-HQ hair transferFID: 18.72 -> 16.23Helped
Use SEEN inpaint instead of the Barbershop optimization replacement (ablation)HairFast · CelebA-HQ hair transferFID: 12.79 -> 16.23Hurt
Add Refinement alignment (ablation)HairFast · CelebA-HQ hair transferFID: 16.23 -> 13.12Helped

Surprising findings

Limitations

Experimental setup

Datasets: FFHQ CelebA-HQ

Benchmarks: CelebA-HQ

Metrics: FID FID_CLIP LPIPS PSNR runtime

Baselines: HairCLIP HairCLIPv2 CtrlHair StyleYourHair Barbershop

Builds on: Barbershop CtrlHair HairCLIP StyleGAN E4E SEAN

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy