FUMO

Pretrained weights for FUMO: Prior-Modulated Diffusion for Single Image Reflection Removal.

Paper

• arXiv: https://arxiv.org/abs/2603.19036
• Hugging Face Papers: https://huggingface.co/papers/2603.19036

Code

GitHub repository: https://github.com/Lucious-Desmon/FUMO

Weights Layout

FUMO/
  base_model/
    vae/
    text_encoder/
    tokenizer/
    scheduler/
  diffusion/
    controlnet/
    unet/
  refine/
    nafnet_refine_final.pth
    nafnet_refine_head_final.pth

Download

huggingface-cli download LulululuDe/FUMO --local-dir ./checkpoints/FUMO

Usage

See the GitHub README for installation, prior generation, and inference instructions.

For inference, the main checkpoint arguments correspond to:

--pretrained_model_name_or_path ./checkpoints/FUMO/base_model
--controlnet_dir ./checkpoints/FUMO/diffusion/controlnet
--unet_dir ./checkpoints/FUMO/diffusion/unet
--refine_net_path ./checkpoints/FUMO/refine/nafnet_refine_final.pth
--refine_head_path ./checkpoints/FUMO/refine/nafnet_refine_head_final.pth

Citation

@article{xu2026fumo,
  title={FUMO: Prior-Modulated Diffusion for Single Image Reflection Removal},
  author={Xu, Telang and Zhang, Chaoyang and Zhai, Guangtao and Liu, Xiaohong},
  journal={arXiv preprint arXiv:2603.19036},
  year={2026}
}