R2Inference - TensorFlow-Lite Examples

From RidgeRun Developer Connection
Jump to: navigation, search



Previous: Examples/TensorFlow Index Next: Examples/TensorRT




R2Inference provides several examples to highlight its usage with different architectures and frameworks. The models used for these examples can be downloaded from our Model Zoo.

Preparation

To test the Tensorflow-Lite example you will need a Tensorflow-Lite C++ API compatible graph file. You can obtain the model from the Model Zoo.

Inception v2

This example is located in r2inference/examples/r2i/tflite . To use this example run:

./inception -i [JPG input Image] -m [Tflite Model] -s [Model Input Size]

For example, evaluating this image:

Error creating thumbnail: Unable to save thumbnail to destination
Inception example input

Should produce the following output:

./inception -i Plane_example_input.jpg -m graph_inceptionv2.tflite -s 224
Loading Model: graph_inceptionv2.tflite
Setting model to engine
Loading image: Plane_example_input.jpg
Configuring frame
Starting engine
Predicting...
Highest probability is label 405 (0.704471)
Stopping engine

According to the ImageNet labels 405 corresponds to an 'airship'.



Previous: Examples/TensorFlow Index Next: Examples/TensorRT