diff options
author | Vibhav Aggarwal <v.aggarwal@samsung.com> | 2024-02-01 19:36:31 +0900 |
---|---|---|
committer | Vibhav Aggarwal <v.aggarwal@samsung.com> | 2024-02-02 11:26:18 +0900 |
commit | f02a2299c4866ab0764767bb638a49ea21fd8039 (patch) | |
tree | 0c5a6c84f929d27f84b1599f7a120ccda02f1be3 /mv_machine_learning/object_detection | |
parent | b2a590a8ee4190352e3f637c7758e4e86fb48353 (diff) | |
download | mediavision-f02a2299c4866ab0764767bb638a49ea21fd8039.tar.gz mediavision-f02a2299c4866ab0764767bb638a49ea21fd8039.tar.bz2 mediavision-f02a2299c4866ab0764767bb638a49ea21fd8039.zip |
mv_machine_learning: minor bug fixes
[Issue type] bug fix
Use default model name in case no model name is provided.
Also, fix a typo in PostprocessParser.h.
Change-Id: I82abf52e6c3ca582721a69ffabd1745e8b2e8462
Signed-off-by: Vibhav Aggarwal <v.aggarwal@samsung.com>
Diffstat (limited to 'mv_machine_learning/object_detection')
-rw-r--r-- | mv_machine_learning/object_detection/include/ObjectDetectionAdapter.h | 2 | ||||
-rw-r--r-- | mv_machine_learning/object_detection/src/ObjectDetectionAdapter.cpp | 7 |
2 files changed, 3 insertions, 6 deletions
diff --git a/mv_machine_learning/object_detection/include/ObjectDetectionAdapter.h b/mv_machine_learning/object_detection/include/ObjectDetectionAdapter.h index 098964e4..e62c4b1d 100644 --- a/mv_machine_learning/object_detection/include/ObjectDetectionAdapter.h +++ b/mv_machine_learning/object_detection/include/ObjectDetectionAdapter.h @@ -37,7 +37,7 @@ private: const std::string _config_file_name = "object_detection.json"; const std::string _plugin_config_file_name = "object_detection_plugin.json"; - void create(std::string model_name = ""); + void create(const std::string &model_name = ""); template<typename U> void create(ObjectDetectionTaskType task_type); ObjectDetectionTaskType convertToTaskType(std::string model_name); diff --git a/mv_machine_learning/object_detection/src/ObjectDetectionAdapter.cpp b/mv_machine_learning/object_detection/src/ObjectDetectionAdapter.cpp index 1b7e4c6e..d089f05d 100644 --- a/mv_machine_learning/object_detection/src/ObjectDetectionAdapter.cpp +++ b/mv_machine_learning/object_detection/src/ObjectDetectionAdapter.cpp @@ -61,12 +61,9 @@ template<typename U> void ObjectDetectionAdapter::create(ObjectDetectionTaskType // TODO. } -void ObjectDetectionAdapter::create(string model_name) +void ObjectDetectionAdapter::create(const string &model_name) { - if (model_name.empty()) - model_name = _config->getDefaultModelName(); - - auto task_type = convertToTaskType(model_name); + auto task_type = convertToTaskType(model_name.empty() ? _config->getDefaultModelName() : model_name); if (_config->isPluginUsed()) { const auto &plugin_name = _config->getPluginFileName(); |