PyTorch Implementation of "Image Inpainting for Irregular Holes Using Partial Convolutions"

Описание

This is the final model after training and fine tuning on the Places2 dataset. Currently, it only supports 256x256 images :(, but im working on it...
Full Code: https://github.com/bobqywei/inpainting-partial-conv

A BIG thank you to...
Nvidia Research Team - https://arxiv.org/pdf/1804.07723.pdf
TensorFlow implementation - https://github.com/SeitaroShinagawa/chainer-partial_convolution_image_inpainting