Tensorflow Image Classification

CNN for multi-class image recognition in tensorflow

Notebook converted from Hvass-Labs' tutorial in order to work with custom datasets, flexible image dimensions, 3-channel images, training over epochs, early stopping, and a deeper network. An updated version of the notebook for TensorFlow 2 is also included, along with a separate requirements file for that TensorFlow version.

This example uses Kaggle's cats vs. dogs dataset.