- Install python deps using
pip install -r requirements.txt. - Download
train.zipandtest.zipfrombit.ly/2mgYsqh, extract into a folder nameddata. - Run
create_tfrecords.pyto create training tfrecord. Runslim/train_pc.shto train model. The loss profile should look something like below.
- Run
slim/run_on_test.shto generate predictions on test set. slim/eval_pc.shcan be used to check accuracy on training set or validation set.- The flask app for monitoring is run by
python3 app/app.py. slim/export_frozen_graph.shandslim/export_inference_graph.shto be used for converting saved TF models to frozen graph andtflite_converter.pyto convert to TFLite.slim/train_logshas the exported frozen graph as well theJSONfile with predictions on the test set.- Android app code inside
android.