Skip to content
Snippets Groups Projects
user avatar
Fabien authored
4fcb072e
History

Deep image prior regularized by coupled total variation for image colorization

In this repository we provide Jupyter Notebooks to reproduce the results of :

Deep image prior regularized by coupled total variation for image colorization SSVM 2023 Gaetano Agazzotti, Fabien Pierre and Frédéric Sur paper

The code of the following papers have been used and modified:

Colorful Image Colorization ECCV 2016 Richard Zhang, Phillip Isola, Alexei A. Efros Project Page

Deep Image Prior CVPR 2018 Dmitry Ulyanov, Andrea Vedaldi, Victor Lempitsky Project Page

Install

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

  • python = 3.7
  • jupyter
  • matplotlib = 3.3.4
  • torch = 1.10.1
  • torchvision = 0.11.2
  • scikit-image = 0.17.2
  • numpy = 1.19.5
  • easydict = 1.9

All of them can be installed via pip , e.g.

pip install -r requirements.txt

Citation

@inproceedings{Agazzotti23deep,
author    = {Agazzotti, Gaetano and Pierre, Fabien and Sur, Frédéric},
title     = {Deep image prior regularized by coupled total variation for image colorization},
booktitle   = {Proceedings of Scale Space and Variational Methods in Computer Vision (SSVM)},
year      = {2023}
}