atlas

Research map / Image generation / Image restoration

Relightful Harmonization: Lighting-aware Portrait Background Replacement

CVPR 2024 · Image generation › Image restoration · approach: Reflection removal · 28 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 Relightful Harmonization, a lighting-aware diffusion model that harmonizes portrait foregrounds to arbitrary background images by encoding background lighting, aligning it with environment-map features, and fine-tuning on synthetic real-image pairs.

Problem

Portrait harmonization for background replacement typically adjusts only global color and brightness, ignoring lighting direction and shadows from the background; relighting methods need HDR environment maps that are not available in casual photography.

Approach

A three-stage training of an InstructPix2Pix-initialized diffusion model with a background-derived lighting feature branch, an alignment network mapping background lighting features to environment-map features, and fine-tuning on a synthetic dataset generated from real natural images by the model itself, so no environment map is needed at inference.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
replace the strongest baseline, Harmonizer, with Relightful HarmonizationPortrait harmonization · Light stage test setMSE: 0.015 -> 0.012Helped
replace the strongest baseline, Harmonizer, with Relightful HarmonizationPortrait harmonization · Light stage test setPSNR: 19.304 -> 20.527Helped
replace the strongest baseline, INR, with Relightful HarmonizationPortrait harmonization · Light stage test setSSIM: 0.823 -> 0.848Helped
replace the strongest baseline, PIH, with Relightful HarmonizationPortrait harmonization · Light stage test setLPIPS: 0.330 -> 0.159Helped
replace the strongest baseline, INR, with Relightful HarmonizationPortrait harmonization · Natural image test setMSE: 0.009 -> 0.005Helped
replace the strongest baseline, INR, with Relightful HarmonizationPortrait harmonization · Natural image test setPSNR: 21.566 -> 23.562Helped
replace the strongest baseline, Harmonizer, with Relightful HarmonizationPortrait harmonization · Natural image test setSSIM: 0.905 -> 0.913Helped
replace the strongest baseline, Harmonizer, with Relightful HarmonizationPortrait harmonization · Natural image test setLPIPS: 0.108 -> 0.097Helped
add background-image lighting conditioning (ablation)Relightful Harmonization · Light stage test setPSNR: 18.768 -> 19.748Helped
use environment-map lighting conditioning instead of background-image conditioning (ablation)Relightful Harmonization · Light stage test setPSNR: 19.748 -> 21.626Helped
add lighting representation alignment to background-conditioned lighting (ablation)Relightful Harmonization · Light stage test setPSNR: 19.748 -> 20.439Helped
add photorealism finetuning on synthesized natural-image pairs (ablation)Relightful Harmonization · Natural image test setPSNR: 20.006 -> 23.562Helped

Surprising findings

Limitations

Experimental setup

Datasets: light stage rendered dataset natural images with synthetic pairs

Benchmarks: light stage test set (500 pairs) natural image synthetic test set (200 pairs) real-world portraits

Metrics: MSE PSNR SSIM LPIPS user preference rate

Baselines: INR PCT Harmonizer PIH TR (Total Relighting) Transformer relighting baseline

Builds on: InstructPix2Pix Total Relighting PIH Harmonizer

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