---
license: apache-2.0
tags:
- image-restoration
- reflection-removal
- diffusion
- computer-vision
- arxiv:2603.19036
pipeline_tag: image-to-image
---

# 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

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

## Download

```bash
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:

```bash
--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

```bibtex
@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}
}
```