metadata
base_model:
- stabilityai/stable-diffusion-2
pipeline_tag: image-to-image
library_name: diffusers
tags:
- reflection-removal
- reflection-separation
- image-restoration
- diffusion
- stable-diffusion
- cvpr2026
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:
huggingface-cli download Brian9999/diff-reflection-separation --repo-type model --local-dir ./checkpoints
Citation
@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}
}