EdgeML/examples/tf
Anil Kag 061881b854
tflite converter utilities for ProtoNN and Bonsai inference (#193)
* Adding keras inference layer to be used along with tflite API proposed by Tensorflow in the 2.0 version. This helps to convert the pre-trained ProtoNN and Bonsai models into the tflite format and shows how to use this model for prediction using the tflite interpreter. Note that the model training still happens on the Tensorflow-1.15 version (this can be changed later in which case you'll be left with only the Bonsai and ProtoNN Keras layers). For using these utilities you will need to use the tensorflow-2.X for conversion.

* Added instructions to use the tflite model converter utilities and the Keras layers. Also updated the convert utility to include the quantization as an optimization step enabled by default. This also includes a bug fix in the requirements-gpu file which incorrectly uses an older version of the numpy library incompatible with the tensorflow version 1.15

* Included assertion that the converter utility works only on Tensorflow 2.X API.
2020-07-16 20:28:12 -07:00
..
Bonsai tflite converter utilities for ProtoNN and Bonsai inference (#193) 2020-07-16 20:28:12 -07:00
EMI-RNN EMIRNN example fixes 2019-08-18 06:41:05 +05:30
FastCells Fixing num_weight_matrices in edgeml_pytorch/rnn.py and readmes (#159) 2020-03-02 19:54:36 -08:00
ProtoNN tflite converter utilities for ProtoNN and Bonsai inference (#193) 2020-07-16 20:28:12 -07:00