simple-dip

Simplified version of https://github.com/DmitryUlyanov/deep-image-prior

• So far, only inpainting has been implemented.
• No guarantees for exactly identical implementations.

How to

1. Install PyTorch, NumPy, Pillow and OpenCV (or uncomment the three lines with cv2 if you do not want the preview while training)
2. Run inpaint.py

This will download the library test image and mask from https://github.com/DmitryUlyanov/deep-image-prior and inpaint the masked region.

Example

Masked image	Inpainted image

Citation

@article{UlyanovVL17,
    author    = {Ulyanov, Dmitry and Vedaldi, Andrea and Lempitsky, Victor},
    title     = {Deep Image Prior},
    journal   = {arXiv:1711.10925},
    year      = {2017}
}