"torchvision/csrc/ops/ps_roi_align.h" did not exist on "0ebbb0abd0610c8ffe978902c06751f94a2e3197"
Classifier.h 704 Bytes
Newer Older
yangql's avatar
yangql committed
1
2
3
#ifndef __CLASSIFIER_H__
#define __CLASSIFIER_H__

4
#include <onnxruntime_cxx_api.h>
yangql's avatar
yangql committed
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
#include <CommonDefinition.h>

namespace ortSamples
{
class  Classifier
{
public:
    Classifier();
    
    ~Classifier();

    ErrorCode Initialize(InitializationParameterOfClassifier initializationParameterOfClassifier);

    ErrorCode Classify(const std::vector<cv::Mat> &srcImages,std::vector<std::vector<ResultOfPrediction>> &predictions);

private:
    cv::FileStorage configurationFile;
    Ort::Session *session;
    cv::Size inputSize;
    std::string inputName;
    Ort::Env env = Ort::Env(ORT_LOGGING_LEVEL_ERROR, "ONNXRuntime");
    Ort::SessionOptions sessionOptions = Ort::SessionOptions();
};
}

#endif