https://github.com/DmitryUlyanov/deep-image-prior

Image restoration with neural networks but without learning.
https://github.com/DmitryUlyanov/deep-image-prior

Last synced: about 1 month ago
JSON representation

Image restoration with neural networks but without learning.

• Host: GitHub
• URL: https://github.com/DmitryUlyanov/deep-image-prior
• Owner: DmitryUlyanov
• License: other
• Created: 2017-11-29T22:33:00.000Z (about 7 years ago)
• Default Branch: master
• Last Pushed: 2023-04-27T03:47:44.000Z (over 1 year ago)
• Last Synced: 2024-10-21T11:53:36.031Z (about 2 months ago)
• Language: Jupyter Notebook
• Homepage: https://dmitryulyanov.github.io/deep_image_prior
• Size: 23.7 MB
• Stars: 7,860
• Watchers: 225
• Forks: 1,429
• Open Issues: 70
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

• my-awesome-starred - DmitryUlyanov/deep-image-prior - Image restoration with neural networks but without learning. (Jupyter Notebook)
• awesome-neural-art - deep-image-prior - Image restoration with neural networks but without learning. Does artifact removal, inpainting, super-resolution, denoising. (Multiple categories)

README

        
**Warning!** The optimization may not converge on some GPUs. We've personally experienced issues on Tesla V100 and P40 GPUs. When running the code, make sure you get similar results to the paper first. Easiest to check using text inpainting notebook.  Try to set double precision mode or turn off cudnn. 

# Deep image prior

In this repository we provide *Jupyter Notebooks* to reproduce each figure from the paper:

> **Deep Image Prior**

> CVPR 2018

> Dmitry Ulyanov, Andrea Vedaldi, Victor Lempitsky

[[paper]](https://sites.skoltech.ru/app/data/uploads/sites/25/2018/04/deep_image_prior.pdf) [[supmat]](https://box.skoltech.ru/index.php/s/ib52BOoV58ztuPM) [[project page]](https://dmitryulyanov.github.io/deep_image_prior)

![](data/teaser_compiled.jpg)

Here we provide hyperparameters and architectures, that were used to generate the figures. Most of them are far from optimal. Do not hesitate to change them and see the effect.

We will expand this README with a list of hyperparameters and options shortly.

# Install

Here is the list of libraries you need to install to execute the code:

- python = 3.6

- [pytorch](http://pytorch.org/) = 0.4

- numpy

- scipy

- matplotlib

- scikit-image

- jupyter

All of them can be installed via `conda` (`anaconda`), e.g.

```

conda install jupyter

```

or create an conda env with all dependencies via environment file

```

conda env create -f environment.yml

```

## Docker image

Alternatively, you can use a Docker image that exposes a Jupyter Notebook with all required dependencies. To build this image ensure you have both [docker](https://www.docker.com/) and  [nvidia-docker](https://github.com/NVIDIA/nvidia-docker) installed, then run

```

nvidia-docker build -t deep-image-prior .

```

After the build you can start the container as

```

nvidia-docker run --rm -it --ipc=host -p 8888:8888 deep-image-prior

```

you will be provided an URL through which you can connect to the Jupyter notebook.

## Google Colab

To run it using Google Colab, click [here](https://colab.research.google.com/github/DmitryUlyanov/deep-image-prior) and select the notebook to run. Remember to uncomment the first cell to clone the repository into colab's environment.

# Citation

```

@article{UlyanovVL17,

    author    = {Ulyanov, Dmitry and Vedaldi, Andrea and Lempitsky, Victor},

    title     = {Deep Image Prior},

    journal   = {arXiv:1711.10925},

    year      = {2017}

}

```