Skip to content

myprojectideas/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

Image Classification in Python with TensorFlow

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 99.9%
  • Python 0.1%