mv_machine_learning/image_classification/include/iimage_classification.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 __IIMAGE_CLASSIFICATION_H__
  18 #define __IIMAGE_CLASSIFICATION_H__
  19
  20 #include <mv_common.h>
  21
  22 #include "image_classification_type.h"
  23
  24 namespace mediavision
  25 {
  26 namespace machine_learning
  27 {
  28 class IImageClassification
  29 {
  30 public:
  31         virtual ~IImageClassification() {};
  32
  33         virtual void preDestroy() = 0;
  34         virtual void setEngineInfo(std::string engine_type, std::string device_type) = 0;
  35         virtual void getNumberOfEngines(unsigned int *number_of_engines) = 0;
  36         virtual void getEngineType(unsigned int engine_index, char **engine_type) = 0;
  37         virtual void getNumberOfDevices(const char *engine_type, unsigned int *number_of_devices) = 0;
  38         virtual void getDeviceType(const char *engine_type, const unsigned int device_index, char **device_type) = 0;
  39         virtual void configure() = 0;
  40         virtual void prepare() = 0;
  41         virtual void perform(mv_source_h &mv_src) = 0;
  42         virtual void performAsync(ImageClassificationInput &input) = 0;
  43         virtual ImageClassificationResult &getOutput() = 0;
  44 };
  45
  46 } // machine_learning
  47 } // mediavision
  48
  49 #endif