evertonrobertoauler / tensorflow_image_classifier

TensorFlow Image Classifier Demo by @Sirajology on Youtube

Tensorflow Image Classifier

This is the code for 'Image Classifier in TensorFlow in 5 Min on YouTube. Use this CodeLab by Google as a guide. Also this tutorial is quite helpful.

Requirements

• docker

Usage

You just need to make a "classifier" directory with a directory "data" inside it with all your images For example

 imagens/treinamento/
 imagens/treinamento/data
 imagens/treinamento/data/carro
 imagens/treinamento/data/caminhao
 imagens/treinamento/data/barco

Train process

Just type

 ./train.sh imagens/treinamento

And it will do anything for you !

Guess process

Just type for a single guess

 ./guess.sh imagens/treinamento imagens/teste/1.jpg