jconsidi / wavepaint

Wavelet-based Image InPainting Model

WavePaint

Resource-efficient Token-mixer for Self-supervised Inpainting

[arXiv]

Thick Mask

Medium Mask

Thin Mask

Model Architecture

We have used LaMa training and inference codes for our experiments.

Citation

If you found this code helpful, please consider citing:

@misc{jeevan2023wavepaint,
      title={WavePaint: Resource-efficient Token-mixer for Self-supervised Inpainting}, 
      author={Pranav Jeevan and Dharshan Sampath Kumar and Amit Sethi},
      year={2023},
      eprint={2307.00407},
      archivePrefix={arXiv},
      primaryClass={cs.CV}
}