Sankek / ImageInpainting

Image Inpainting using Partial Convolutions

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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: Open In Colab

Inpainting example Inpainting example Inpainting example

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:

Mask examples

Training notebook: Open In Colab

About

Image Inpainting using Partial Convolutions

License:MIT License


Languages

Language:Jupyter Notebook 97.2%Language:Python 2.8%