R2Inference - TensorFlow-Lite Examples

From RidgeRun Developer Wiki




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:

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