init commit

include/nn/onnx_model_base.h

@@ -0,0 +1,39 @@
+#pragma once
+#include <onnxruntime_cxx_api.h>
+#include <string>
+#include <unordered_map>
+#include <vector>
+
+/*
+ * This interface must provide only required arguments to load any onnx model regarding specific info -
+ *  - i.e. modelPath will always be required, provider like "cpu" or "cuda" the same, since these are parameters you need
+ *  to set up `sessionOptions` or `session` objects properly, but image size is not needed for pure onnx graph to be loaded so do NOT include it here
+ */
+class OnnxModelBase {
+public:
+    OnnxModelBase(const char* modelPath, const char* logid, const char* provider);
+    //OnnxModelBase();  // no default constructor should be there
+    //virtual ~OnnxModelBase();
+    virtual const std::vector<std::string>& getInputNames(); // = 0
+    virtual const std::vector<std::string>& getOutputNames();
+    virtual const std::vector<const char*> getOutputNamesCStr();
+    virtual const std::vector<const char*> getInputNamesCStr();
+    virtual const Ort::ModelMetadata& getModelMetadata();
+    virtual const std::unordered_map<std::string, std::string>& getMetadata();
+    virtual const char* getModelPath();
+    virtual const Ort::Session& getSession();
+    //virtual std::vector<Ort::Value> forward(std::vector<Ort::Value> inputTensors);
+    virtual std::vector<Ort::Value> forward(std::vector<Ort::Value>& inputTensors);
+    Ort::Session session{ nullptr };
+
+protected:
+    const char* modelPath_;
+    Ort::Env env{ nullptr };
+
+    std::vector<std::string> inputNodeNames;
+    std::vector<std::string> outputNodeNames;
+    Ort::ModelMetadata model_metadata{ nullptr };
+    std::unordered_map<std::string, std::string> metadata;
+    std::vector<const char*> outputNamesCStr;
+    std::vector<const char*> inputNamesCStr;
+};