Awesome
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