auracaster/piper
3
0
Fork 0
mirror of https://github.com/pstrueb/piper.git synced 2026-05-19 03:58:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

Upgrade to onnxruntime 1.13.1

Browse Source
This commit is contained in:
Michael Hansen
2023-01-14 12:09:23 -06:00
parent afbcfb4df3
commit b2af638072
15 changed files with 16 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/cpp/model.hpp
View File

@@ -12,8 +12,6 @@ const string instanceName{"larynx"};
struct ModelSession {
Ort::Session onnx;
vector<char *> inputNames;
vector<char *> outputNames;
Ort::AllocatorWithDefaultOptions allocator;
Ort::SessionOptions options;
Ort::Env env;
@@ -48,19 +46,6 @@ void loadModel(string modelPath, ModelSession &session) {
session.onnx = Ort::Session(session.env, modelPath.c_str(), session.options);
auto endTime = chrono::steady_clock::now();
auto loadDuration = chrono::duration<double>(endTime - startTime);
size_t numInputNodes = session.onnx.GetInputCount();
size_t numOutputNodes = session.onnx.GetOutputCount();
for (size_t i = 0; i < numInputNodes; i++) {
session.inputNames.push_back(
session.onnx.GetInputName(i, session.allocator));
}
for (size_t i = 0; i < numOutputNodes; i++) {
session.outputNames.push_back(
session.onnx.GetOutputName(i, session.allocator));
}
}
} // namespace larynx