/tensorflow-image-classification

CNN for multi-class image recognition in tensorflow

Primary LanguageJupyter Notebook

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.

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