/image-classification-demo

Demonstrate image classification using machine learning

Primary LanguagePython

image-classification-demo

Demonstrate image classification using machine learning

What is this repo?

I used teachable machine to train a machine learning model to classify images as cats or dogs. The slideshow gives an overview of the process.

There is a python script in the code directory which can run the models created using teachable machine, which have been exported to the keras format.

How to run

cd to the code directory, then run python classify_images.py original_keras_model.h5.