From 5d2be0054cd01932392a3bf8bb77f0bbfa0ad300 Mon Sep 17 00:00:00 2001 From: Inki Dae Date: Fri, 13 May 2022 16:42:36 +0900 Subject: [PATCH] update backend name if backend type is given [Version] : 0.4.2-0 [Issue type] : bug fix Change-Id: I45c1e909ebde6ffe855d8e702898584841fe38e6 Signed-off-by: Inki Dae --- packaging/inference-engine-interface.spec | 2 +- src/inference_engine_common_impl.cpp | 12 ++++++++++++ 2 files changed, 13 insertions(+), 1 deletion(-) diff --git a/packaging/inference-engine-interface.spec b/packaging/inference-engine-interface.spec index c1545ba..caf63f2 100644 --- a/packaging/inference-engine-interface.spec +++ b/packaging/inference-engine-interface.spec @@ -1,6 +1,6 @@ Name: inference-engine-interface Summary: Interface of inference engines -Version: 0.4.1 +Version: 0.4.2 Release: 0 Group: Multimedia/Framework License: Apache-2.0 diff --git a/src/inference_engine_common_impl.cpp b/src/inference_engine_common_impl.cpp index f3c9f75..3467ee0 100644 --- a/src/inference_engine_common_impl.cpp +++ b/src/inference_engine_common_impl.cpp @@ -391,6 +391,18 @@ namespace Common BackendTable.insert(std::make_pair("snpe", INFERENCE_BACKEND_SNPE)); config->backend_type = BackendTable[config->backend_name]; + } else { + std::map BackendTable; + + BackendTable.insert(std::make_pair(INFERENCE_BACKEND_TFLITE, "tflite")); + BackendTable.insert(std::make_pair(INFERENCE_BACKEND_ARMNN, "armnn")); + BackendTable.insert(std::make_pair(INFERENCE_BACKEND_OPENCV, "opencv")); + BackendTable.insert(std::make_pair(INFERENCE_BACKEND_MLAPI, "mlapi")); + BackendTable.insert(std::make_pair(INFERENCE_BACKEND_ONE, "one")); + BackendTable.insert(std::make_pair(INFERENCE_BACKEND_NNTRAINER, "nntrainer")); + BackendTable.insert(std::make_pair(INFERENCE_BACKEND_SNPE, "snpe")); + + config->backend_name = BackendTable[config->backend_type]; } std::string backendLibName; -- 2.34.1