CIFAR-10 image classfier using TensorFlow

This design is a small convolution neural network that is trained to identify the 10 image classes of the CIFAR-10 dataset. The dataset is downloaded using the native Keras dataset functions, then trained and evaluated using TensorFlow.

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 tf.layers API.
  • Logging of data for display in TensorBoard.
  • Saving the model graph as a protobuf text file.


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