Skip to content

BaseMax/ColorizeBlackWhiteImagePython

Repository files navigation

Colorize Black&White Images Python

Colorize black & white images, using machine learning in Python.

Original Image Gray image Colorized image by app

Model

  • colorization_release_v2.caffemodel: It is a pre-trained model stored in the Caffe framework’s format that can be used to predict new unseen data.
  • colorization_deploy_v2.prototxt: It consists of different parameters that define the network and it also helps in deploying the Caffe model.
  • pts_in_hull.npy: It is a NumPy file that stores the cluster center points in NumPy format. It consists of 313 cluster kernels, i.e (0-312).

For downloading the files, click here.

Sources

Note: This is not mine and I used another model which is available on the internet. For reading more you can look at following links:

Thanks all and also great OpenCV lib.