framework/phool/onnxlib.h

0001 #ifndef PHOOL_ONNXLIB_H
0002 #define PHOOL_ONNXLIB_H
0003
0004 #include <onnxruntime_c_api.h>
0005 #include <onnxruntime_cxx_api.h>
0006 // This is a stub for some ONNX code refactoring
0007
0008 Ort::Session *onnxSession(std::string &modelfile, int verbosity = 0);
0009
0010 std::vector<float> onnxInference(Ort::Session *session, std::vector<float> &input, int N, int Nsamp, int Nreturn);
0011
0012 std::vector<float> onnxInference(Ort::Session *session, std::vector<float> &input, int N, int Nx, int Ny, int Nz, int Nreturn);
0013
0014 namespace onnxlib
0015 {
0016   extern int n_input;
0017   extern int n_output;
0018 }  // namespace onnxlib
0019
0020 #endif