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

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

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.

Implement with PyTorch1.0

Name		Name	Last commit message	Last commit date
Latest commit History 19 Commits
.idea		.idea
pt1		pt1
readme images		readme images
README.md		README.md
colornet.py		colornet.py
myimgfolder.py		myimgfolder.py
removegray.py		removegray.py
train.py		train.py
val.py		val.py