https://github.com/richzhang/colorization

Automatic colorization using deep neural networks. "Colorful Image Colorization." In ECCV, 2016.
https://github.com/richzhang/colorization

automatic-colorization caffe colorization computer-vision deep-learning deep-learning-algorithms

Last synced: 21 days ago
JSON representation

Automatic colorization using deep neural networks. "Colorful Image Colorization." In ECCV, 2016.

• Host: GitHub
• URL: https://github.com/richzhang/colorization
• Owner: richzhang
• License: bsd-2-clause
• Created: 2016-03-26T02:33:33.000Z (over 8 years ago)
• Default Branch: master
• Last Pushed: 2023-11-27T04:46:28.000Z (11 months ago)
• Last Synced: 2024-10-01T14:02:53.215Z (about 1 month ago)
• Topics: automatic-colorization, caffe, colorization, computer-vision, deep-learning, deep-learning-algorithms
• Language: Python
• Homepage: http://richzhang.github.io/colorization/
• Size: 174 MB
• Stars: 3,335
• Watchers: 117
• Forks: 925
• Open Issues: 52
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

• awesome-neural-art - Colorful Image Colorization
• Awesome-CoreML-Models - Colorization

README

        

## Colorful Image Colorization [[Project Page]](http://richzhang.github.io/colorization/) 


[Richard Zhang](https://richzhang.github.io/), [Phillip Isola](http://web.mit.edu/phillipi/), [Alexei A. Efros](http://www.eecs.berkeley.edu/~efros/). In [ECCV, 2016](http://arxiv.org/pdf/1603.08511.pdf).

**+ automatic colorization functionality for Real-Time User-Guided Image Colorization with Learned Deep Priors, SIGGRAPH 2017!**

**[Sept20 Update]** Since it has been 3-4 years, I converted this repo to support minimal test-time usage in PyTorch. I also added our SIGGRAPH 2017 (it's an interactive method but can also do automatic). See the [Caffe branch](https://github.com/richzhang/colorization/tree/caffe) for the original release.

![Teaser Image](http://richzhang.github.io/colorization/resources/images/teaser4.jpg)

**Clone the repository; install dependencies**

```

git clone https://github.com/richzhang/colorization.git

pip install requirements.txt

```

**Colorize!** This script will colorize an image. The results should match the images in the `imgs_out` folder.

```

python demo_release.py -i imgs/ansel_adams3.jpg

```

**Model loading in Python** The following loads pretrained colorizers. See [demo_release.py](demo_release.py) for some details on how to run the model. There are some pre and post-processing steps: convert to Lab space, resize to 256x256, colorize, and concatenate to the original full resolution, and convert to RGB.

```python

import colorizers

colorizer_eccv16 = colorizers.eccv16().eval()

colorizer_siggraph17 = colorizers.siggraph17().eval()

```

### Original implementation (Caffe branch)

The original implementation contained train and testing, our network and AlexNet (for representation learning tests), as well as representation learning tests. It is in Caffe and is no longer supported. Please see the [caffe](https://github.com/richzhang/colorization/tree/caffe) branch for it.

### Citation ###

If you find these models useful for your resesarch, please cite with these bibtexs.

```

@inproceedings{zhang2016colorful,

  title={Colorful Image Colorization},

  author={Zhang, Richard and Isola, Phillip and Efros, Alexei A},

  booktitle={ECCV},

  year={2016}

}

@article{zhang2017real,

  title={Real-Time User-Guided Image Colorization with Learned Deep Priors},

  author={Zhang, Richard and Zhu, Jun-Yan and Isola, Phillip and Geng, Xinyang and Lin, Angela S and Yu, Tianhe and Efros, Alexei A},

  journal={ACM Transactions on Graphics (TOG)},

  volume={9},

  number={4},

  year={2017},

  publisher={ACM}

}

```

### Misc ###

Contact Richard Zhang at rich.zhang at eecs.berkeley.edu for any questions or comments.