From c89e41baf23bf2d32d25fc9f996e9c18ded212ba Mon Sep 17 00:00:00 2001 From: tvlenin Date: Mon, 10 Feb 2020 15:26:14 -0600 Subject: [PATCH 1/2] Fix tflite prediction size --- r2i/tflite/engine.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/r2i/tflite/engine.cc b/r2i/tflite/engine.cc index 0a63b7fd..3bdd99eb 100644 --- a/r2i/tflite/engine.cc +++ b/r2i/tflite/engine.cc @@ -201,7 +201,7 @@ std::shared_ptr Engine::Predict (std::shared_ptr int output = this->interpreter->outputs()[0]; TfLiteIntArray *output_dims = this->interpreter->tensor(output)->dims; - auto output_size = output_dims->data[output_dims->size - 1]; + auto output_size = output_dims->data[output_dims->size - 1] * sizeof(float); auto *tensor_data = this->interpreter->typed_output_tensor(0); prediction->SetTensorValues(tensor_data, output_size); From 77aa776a8d441f8aa5baf7cbb39f0326fb743af5 Mon Sep 17 00:00:00 2001 From: tvlenin Date: Mon, 10 Feb 2020 15:59:07 -0600 Subject: [PATCH 2/2] Increase version number --- configure.ac | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/configure.ac b/configure.ac index 07813e45..229f71f6 100644 --- a/configure.ac +++ b/configure.ac @@ -10,7 +10,7 @@ # Initialize autoconf. AC_PREREQ([2.69]) -AC_INIT([RidgeRun inference library],[0.4.0],[https://github.com/RidgeRun/r2inference/issues],[r2inference]) +AC_INIT([RidgeRun inference library],[0.4.1],[https://github.com/RidgeRun/r2inference/issues],[r2inference]) # Initialize our build utils RR_INIT