Skip to content

Fast Colorization of Grayscale Images by Convolutional Neural Network

Notifications You must be signed in to change notification settings

rubyruins/fast-colorization

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Fast Colorization 🎨

Fast Colorization of Grayscale Images by Convolutional Neural Network.

This model uses interconversion between RGB and LAB color spaces to find traits that link each grayscale value to corresponding colour values. Currently, it has only been trained on a dataset of 700 human faces with accuracy ~ 60%.


Initial results:

image


Future scope:

  • Improve the network to provide colorization for not only for faces but also objects.
  • Train the network with a larger dataset to obtain better results and improve accuracy.

You are welcome to open pull requests :)

About

Fast Colorization of Grayscale Images by Convolutional Neural Network

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages