README.md

LogoClassification

This project shows how to train and predict 3 classes of logo with keras and tensorflow lite

Dependencies

python3, keras 2.2.4, tensorflow 1.15.0

Train your images

Replace the current image folders in dataset/raw-img/train and dataset/raw-img/validation by your image folders.
Then

python3 train.py

Convert your model to tflite

python3 convert2tflite.py

Inference with .h5

python3 predict.py

Inference with .tflite

python3 inference_on_tflite.py

Application example