/CMT

Continuously Masked Transformer for Image Inpainting, ICCV, 2023

Primary LanguagePython

Python 3.7

CMT

Keunsoo Ko and Chang-Su Kim

Official PyTorch Code for "Continuously Masked Transformer for Image Inpainting, ICCV, 2023"

Installation

Download repository:

    $ git clone https://github.com/keunsoo-ko/CMT.git

Download pre-trained model on Places2 or pre-trained model on CelebA

Usage

Run Test:

    $ python demo.py --ckpt CMT.pth(put downloaded model path) --img_path ./samples/test_img --mask_path ./samples/test_mask --output_path ./samples/results