mv_machine_learning/landmark_detection/include/pose_landmark_adapter.h

   1 /**
   2  * Copyright (c) 2023 Samsung Electronics Co., Ltd All Rights Reserved
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  * http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 #ifndef __POSE_LANDMARK_ADAPTER_H__
  18 #define __POSE_LANDMARK_ADAPTER_H__
  19
  20 #include <dlog.h>
  21
  22 #include "EngineConfig.h"
  23 #include "itask.h"
  24 #include "pld_cpm.h"
  25
  26 namespace mediavision
  27 {
  28 namespace machine_learning
  29 {
  30 template<typename T, typename V> class PoseLandmarkAdapter : public mediavision::common::ITask<T, V>
  31 {
  32 private:
  33         std::unique_ptr<LandmarkDetection> _landmark_detection;
  34         T _source;
  35         std::string _model_name;
  36         std::string _model_file;
  37         std::string _meta_file;
  38         std::string _label_file;
  39         const std::string _config_file_name = "pose_landmark.json";
  40
  41         void create(LandmarkDetectionTaskType task_type);
  42         LandmarkDetectionTaskType convertToTaskType(std::string model_name);
  43
  44 public:
  45         PoseLandmarkAdapter();
  46         ~PoseLandmarkAdapter();
  47
  48         void setModelInfo(const char *model_file, const char *meta_file, const char *label_file,
  49                                           const char *model_name) override;
  50         void setEngineInfo(const char *engine_type, const char *device_type) override;
  51         void configure() override;
  52         void getNumberOfEngines(unsigned int *number_of_engines) override;
  53         void getEngineType(unsigned int engine_index, char **engine_type) override;
  54         void getNumberOfDevices(const char *engine_type, unsigned int *number_of_devices) override;
  55         void getDeviceType(const char *engine_type, unsigned int device_index, char **device_type) override;
  56         void prepare() override;
  57         void setInput(T &t) override;
  58         void perform() override;
  59         void performAsync(T &t) override;
  60         V &getOutput() override;
  61         V &getOutputCache() override;
  62 };
  63
  64 } // machine_learning
  65 } // mediavision
  66
  67 #endif