diff options
Diffstat (limited to 'mv_machine_learning/object_detection/include/ObjectDetectionAdapter.h')
-rw-r--r-- | mv_machine_learning/object_detection/include/ObjectDetectionAdapter.h | 66 |
1 files changed, 66 insertions, 0 deletions
diff --git a/mv_machine_learning/object_detection/include/ObjectDetectionAdapter.h b/mv_machine_learning/object_detection/include/ObjectDetectionAdapter.h new file mode 100644 index 00000000..098964e4 --- /dev/null +++ b/mv_machine_learning/object_detection/include/ObjectDetectionAdapter.h @@ -0,0 +1,66 @@ +/** + * Copyright (c) 2022 Samsung Electronics Co., Ltd All Rights Reserved + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +#ifndef __OBJECT_DETECTION_ADAPTER_H__ +#define __OBJECT_DETECTION_ADAPTER_H__ + +#include <dlog.h> + +#include "EngineConfig.h" +#include "ITask.h" +#include "MvMlConfig.h" +#include "MobilenetV1Ssd.h" +#include "MobilenetV2Ssd.h" + +namespace mediavision +{ +namespace machine_learning +{ +class ObjectDetectionAdapter : public mediavision::common::ITask +{ +private: + std::unique_ptr<IObjectDetection> _object_detection; + std::shared_ptr<Config> _config; + 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 = ""); + template<typename U> void create(ObjectDetectionTaskType task_type); + ObjectDetectionTaskType convertToTaskType(std::string model_name); + +public: + ObjectDetectionAdapter(); + ~ObjectDetectionAdapter(); + + void setModelInfo(const std::string &model_file, const std::string &meta_file, const std::string &label_file, + const std::string &model_name) override; + void setEngineInfo(const std::string &engine_type, const std::string &device_type) override; + void configure() override; + unsigned int getNumberOfEngines() override; + const std::string &getEngineType(unsigned int engine_index) override; + unsigned int getNumberOfDevices(const std::string &engine_type) override; + const std::string &getDeviceType(const std::string &engine_type, unsigned int device_index) override; + void prepare() override; + void perform(InputBaseType &input) override; + void performAsync(InputBaseType &input) override; + OutputBaseType &getOutput() override; + OutputBaseType &getOutputCache() override; +}; + +} // machine_learning +} // mediavision + +#endif
\ No newline at end of file |