mv_machine_learning/face_recognition/include/backbone_model_info.h

   1 /**
   2  * Copyright (c) 2022 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 __BACKBONE_MODEL_INFO_H__
  18 #define __BACKBONE_MODEL_INFO_H__
  19
  20 #include "inference_engine_common_impl.h"
  21
  22 typedef struct {
  23         std::string layer_name;
  24         inference_engine_tensor_info tensor_info;
  25 } model_layer_info;
  26
  27 class IBackboneModelInfo
  28 {
  29 public:
  30         virtual ~IBackboneModelInfo()
  31         {}
  32
  33         virtual std::vector<model_layer_info> &GetInputLayerInfo() = 0;
  34         virtual std::vector<model_layer_info> &GetOutputLayerInfo() = 0;
  35         virtual std::string GetModelFilePath() = 0;
  36 };
  37
  38 #endif