pandinosaurus / colornet-pytorch

A Neural Network For Automatic Image Colorization

colorNet-pytorch

A Neural Network For Automatic Image Colorization

This project is a PyTorch version of the ColorNet issued on SIGGRAPH 2016. Please check out the original website for the details.

Overview

• Net model

• DataSet MIT Places205

Hint: For there are grayscale images in the dataset, I write a script to remove these images

• Development Environment
  Python 3.5.1
  CUDA 8.0

Result

I just train this model for 3 epochs while 11 epochs in the paper， so I think it will work better if train it more.

• Good results
  
• Bad results
  
  For this network is trained by landscape image database, it's work well for scenery pictures. So if you use this network to color  images of other types, maybe you can't get a satisfying output.

Pretrained model

You can download the model from https://drive.google.com/file/d/0B6WuMuYfgb4XblE4c3N2RUJQcFU/view?usp=sharing

Todo

Implement with PyTorch1.0