Image Inpainting

This is my implementation of Image Inpainting for Irregular Holes Using Partial Convolutions with slight differences.

Results

Here are the results of training for about 2 epochs on ImageNet.

Draw your mask and test the model by yourself:

Training

The model was trained on ImageNet dataset on 256x256 images.

For training I generate irregular masks with random walk and they look like this:

Training notebook: