# Colorization

See the Jupyter Notebook "Image_Colorization.ipynb" for general usage info. The main colorization algorithm is based on https://github.com/richzhang/colorization. 

## Download training images

Run "models/colored_postcards/download_color.py" to download training images in 256 × 256 pixel resolution

## Download trained models

We host three colorizer models on our webserver:
- <a href='https://labs.onb.ac.at/static/media/colorize/colorization_release_v2-9b330a0b.pth'>ECCV16</a> (by R. Zhang et al., originally downloaded from https://colorizers.s3.us-east-2.amazonaws.com/colorization_release_v2-9b330a0b.pth)
- <a href='https://labs.onb.ac.at/static/media/colorize/siggraph17-df00044c.pth'>SIGGRAPH17</a> (by R. Zhang et al., originally downloaded from https://colorizers.s3.us-east-2.amazonaws.com/siggraph17-df00044c.pth)
- <a href='https://labs.onb.ac.at/static/media/colorize/col_new.pth'>Our trained model</a>

