deep-image-prior

Image restoration with neural networks but without learning.

  • 所有者: DmitryUlyanov/deep-image-prior
  • 平台:
  • 許可證: Other
  • 分類:
  • 主題:
  • 喜歡:
    0
      比較:

Github星跟蹤圖

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

(https://sites.skoltech.ru/app/data/uploads/sites/25/2018/04/deep_image_prior.pdf) (https://box.skoltech.ru/index.php/s/ib52BOoV58ztuPM) (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 = 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 and 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.

Citation

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

主要指標

概覽
名稱與所有者DmitryUlyanov/deep-image-prior
主編程語言Jupyter Notebook
編程語言Jupyter Notebook (語言數: 3)
平台
許可證Other
所有者活动
創建於2017-11-29 22:33:00
推送於2023-04-27 03:47:44
最后一次提交2020-05-09 13:57:11
發布數0
用户参与
星數8k
關注者數227
派生數1.4k
提交數61
已啟用問題?
問題數98
打開的問題數67
拉請求數7
打開的拉請求數3
關閉的拉請求數5
项目设置
已啟用Wiki?
已存檔?
是復刻?
已鎖定?
是鏡像?
是私有?