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.
-
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
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.