Train a neural model to classify images

I recommend running on a python virtualenv, just install the requirements and you are good to go!

First you run the code below to download the dataset to your local machine

python load_dataset.py

Then, train and save your model

python train.py

Finally, load the saved model and start to make predictions

python load_saved_model.py

The model is saved as model_image_classifier.h5.

Enjoy!

quindai/neuralimagemodel