Skip to content

Latest commit

 

History

History
6 lines (4 loc) · 663 Bytes

README.md

File metadata and controls

6 lines (4 loc) · 663 Bytes

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.