Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/DmitryUlyanov/deep-image-prior
Image restoration with neural networks but without learning.
https://github.com/DmitryUlyanov/deep-image-prior
Last synced: 6 days 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 (almost 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 (13 days 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)
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}
}
```