Skip to content

rdcolema/tensorflow-image-classification

Repository files navigation

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.

About

CNN for multi-class image recognition in tensorflow

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published