CIFAR-10 image classfier using Keras

This design is very similar to the TensorFlow version, but uses Keras exclusively.

The example demonstrates the following:

  • Simple download of CIFAR-10 dataset.
  • Scaling of the image data, one-hot encoding of the labels.
  • How to define a CNN using the Keras sequential model
  • How to set up Keras callbacks for Tensorboard logging, early stopping and checkpoint saving.
  • How to compile, train, evaluate the model.
  • How to make predictions with the trained model.
  • How to save the Keras model as HDF5 and JSON formats.
  • How to freeze the graph to a protobuf file from Keras.


The code and a Jupyter notebook is available on my Github page.

If you have any questions or comments about this design, please email me at designs@markharvey.info