Modelpublic

Brian9999/diff-reflection-separation

sourceHugging Faceupdated 4mo agoView on Hugging Face

0likes34downloads

Model Card

Reflection Separation from a Single Image via Joint Latent Diffusion

Pre-trained checkpoints for the CVPR 2026 paper. Given a single photo taken through glass, the model jointly generates the transmission and reflection layers.

—🌐 Project page: https://brian90709.github.io/diff-reflection-separation/
—💻 Code: https://github.com/Brian90709/diff-reflection-separation-code
—📄 Paper: arXiv (coming soon)

Zheng-Hui Huang, Zhixiang Wang, Yu-Lun Liu, Yung-Yu Chuang

Files

File	Size	Description
`iter_016000/unet/diffusion_pytorch_model.bin`	~3.5 GB	Trained layer-separation UNet.
`fuse_blocks.bin`	~264 MB	CFW refiner for the VAE decoder.
`lrm/iter_008000/aux_net.bin`	~1.3 MB	Latent composition module (LRM), used by `--optimization`.

Usage

Download the weights into ./checkpoints, then follow the code repository:

bash

huggingface-cli download Brian9999/diff-reflection-separation --repo-type model --local-dir ./checkpoints

Citation

bibtex

@inproceedings{huang2026reflection,
  title     = {Reflection Separation from a Single Image via Joint Latent Diffusion},
  author    = {Huang, Zheng-Hui and Wang, Zhixiang and Liu, Yu-Lun and Chuang, Yung-Yu},
  booktitle = {CVPR},
  year      = {2026}
}