add nntrainer backend support

[Version] : 0.1.0-0
[Issue type] : new feature

Change-Id: I2dde532dc012d066e1ad7b5060ec93b40d781cef
Signed-off-by: Inki Dae <inki.dae@samsung.com>

diff --git a/include/inference_engine_type.h b/include/inference_engine_type.h
index f854452..b2d48ce 100644 (file)
--- a/include/inference_engine_type.h
+++ b/include/inference_engine_type.h
@@ -44,6 +44,7 @@ extern "C"
                INFERENCE_BACKEND_ARMNN, /**< ARMNN */
                INFERENCE_BACKEND_MLAPI, /** < ML Single API of NNStreamer.*/
                INFERENCE_BACKEND_ONE, /** < On-device Neural Engine. */
+               INFERENCE_BACKEND_NNTRAINER, /** < NNTrainer. */
                INFERENCE_BACKEND_MAX /**< Backend MAX */
        } inference_backend_type_e;
 

diff --git a/packaging/inference-engine-interface.spec b/packaging/inference-engine-interface.spec
index 4ec1b4f..1f46e10 100644 (file)
--- a/packaging/inference-engine-interface.spec
+++ b/packaging/inference-engine-interface.spec
@@ -1,7 +1,7 @@
 Name:        inference-engine-interface
 Summary:     Interface of inference engines
-Version:     0.0.3
-Release:     14
+Version:     0.1.0
+Release:     0
 Group:       Multimedia/Framework
 License:     Apache-2.0
 Source0:     %{name}-%{version}.tar.gz

diff --git a/src/inference_engine_common_impl.cpp b/src/inference_engine_common_impl.cpp
index 5f709a2..a5977aa 100644 (file)
--- a/src/inference_engine_common_impl.cpp
+++ b/src/inference_engine_common_impl.cpp
@@ -99,6 +99,7 @@ namespace Common
                if (backend_type == INFERENCE_BACKEND_MLAPI ||
                                device_type == INFERENCE_TARGET_CUSTOM ||
                                backend_type == INFERENCE_BACKEND_ONE ||
+                               backend_type == INFERENCE_BACKEND_NNTRAINER ||
                                (backend_type == INFERENCE_BACKEND_TFLITE && sApiFwForTFLITE == INFERENCE_BACKEND_MLAPI) ||
                                (backend_type == INFERENCE_BACKEND_ARMNN && sApiFwForARMNN == INFERENCE_BACKEND_MLAPI))
                        return 1;
@@ -382,6 +383,7 @@ namespace Common
                        BackendTable.insert(std::make_pair("opencv",INFERENCE_BACKEND_OPENCV));
                        BackendTable.insert(std::make_pair("mlapi",INFERENCE_BACKEND_MLAPI));
                        BackendTable.insert(std::make_pair("one",INFERENCE_BACKEND_ONE));
+                       BackendTable.insert(std::make_pair("nntrainer", INFERENCE_BACKEND_NNTRAINER));
 
                        config->backend_type = BackendTable[config->backend_name];
                }