The server code for our generative inpainting model. The client code can be found on our inpainting-client repository. It can also be tested online on https://ndarr.github.io/image-inpainting-client/.
We build the model based on the U-Net architecture by Ronneberger et al. with the alterations suggested by Lui et al.. The model was trained on public domain cat pictures. The model parameter file can be found on Google Drive.
- Nicolas Darr - ndarr
- Phillip Rust - xplip
- Alexander Müller - AlexanderMlr
- Thanks to NVIDIA for providing the partialconv2d module for PyTorch: https://github.com/NVIDIA/partialconv
- Thanks to Alexandre Milesi for providing the PyTorch UNet implementation: https://github.com/milesial/Pytorch-UNet