Image Inpainting using Partial Convolutions with PyTorch

this repository is unofficial implementation of Image Inpainting for Irregular Holes Using Partial Convolutions [Liu+, ECCV 2018].

Official implementation is here.

Requirements

• Python3.x
• PyTorch 1.5.0
• pillow
• matplotlib

Usage

• set images under ./img and mask images under ./mask. Then,

python3 train.py

References

• Image Inpainting for Irregular Holes Using Partial Convolutions, Guilin Liu, Fitsum A. Reda, Kevin J. Shih, Ting-Chun Wang, Andrew Tao, Bryan Catanzaro, NVIDIA Corporation, [arXiv]