Beautiful inpainting How to use Pytorch-MADF

Top quality model

input

mask

result

Paper with code places2 A model that ranks fourth after Lama in the inpainting category (as of January 2023).

GitHub - MADF-inpainting/Pytorch-MADF

Usage

install

git clone https://github.com/MADF-inpainting/Pytorch-MADF.git
cd Pytorch-MADF

Download pretrained model weights from the repository.

3 models of laces2 Celeba PSV (all 512px*512px) are provided.

data

Prepare a list file of images and masks as shown below.

examples/places2/case1.png examples/places2/case1_mask.png
examples/places2/case2.png examples/places2/case2_mask.png
examples/places2/case3.png examples/places2/case3_mask.png
examples/places2/case4.png examples/places2/case4_mask.png
examples/places2/case5.png examples/places2/case5_mask.png
examples/places2/case6.png examples/places2/case6_mask.png
examples/places2/case7.png examples/places2/case7_mask.png

python test.py --list_file places2_example_list --snapshot places2.pth

The result is saved in result.

🐣

I’m a freelance engineer.
Work consultation
Please feel free to contact us with a brief development description.
rockyshikoku@gmail.com

I am making an app that uses Core ML and ARKit.
We send machine learning / AR related information.

GitHub

Twitter
Medium