Image Colorizer

Get this modelTry the API


This model is a Generative Adversarial Network (GAN) that was trained by the IBM CODAIT Team on COCO dataset images converted to grayscale and produces colored images. The input to the model is a grayscale image (jpeg or png), and the output is a colored 256 by 256 image (increased resolution will be added in future releases). The model is based on Christopher Hesse’s Tensorflow implementation of the pix2pix model.

Model Metadata

Domain Application Industry Framework Training Data Input Data Format
Vision Image Coloring General TensorFlow COCO Dataset Images



Component License Link
Model GitHub Repository Apache 2.0 LICENSE
Model Code (3rd party) MIT TensorFlow pix2pix Repository
Model Weights Apache 2.0 LICENSE
Test Assets CC0 License Asset README

Options available for deploying this model

  • Deploy from Dockerhub:
docker run -it -p 5000:5000 codait/max-image-colorizer
  • Deploy on Kubernetes:
kubectl apply -f

Example Usage

Once deployed you can upload an image using the UI:

Swagger Doc Screenshot

You can also test the model from the command line. For example:

curl -F "image=@assets/bw-city.jpg" -XPOST http://localhost:5000/model/predict > result.png && open result.png