summaryrefslogtreecommitdiffhomepage
diff options
context:
space:
mode:
authorMichael Welter <DaWelter@users.noreply.github.com>2021-10-24 09:10:52 +0200
committerGitHub <noreply@github.com>2021-10-24 09:10:52 +0200
commit1044c1d3cda8c0cd57447f95138498e5d7045b04 (patch)
tree806788f669e51090e47e22de9ef483ed0deabefa
parentb3880a052ef103fa9d171158c1ea38a51c220cbd (diff)
parent6b89c27a5eb681b4403969dce898b74947071a99 (diff)
Merge pull request #1358 from DaWelter/master
tracker/nn: Remove onnx session options which hurt performance on Linux
-rw-r--r--tracker-neuralnet/ftnoir_tracker_neuralnet.cpp6
1 files changed, 1 insertions, 5 deletions
diff --git a/tracker-neuralnet/ftnoir_tracker_neuralnet.cpp b/tracker-neuralnet/ftnoir_tracker_neuralnet.cpp
index 8beb4119..f5939cf6 100644
--- a/tracker-neuralnet/ftnoir_tracker_neuralnet.cpp
+++ b/tracker-neuralnet/ftnoir_tracker_neuralnet.cpp
@@ -568,16 +568,12 @@ bool neuralnet_tracker::load_and_initialize_model()
// before running the inference pass.
opts.SetIntraOpNumThreads(num_threads);
opts.SetInterOpNumThreads(num_threads);
- opts.SetGraphOptimizationLevel(
- GraphOptimizationLevel::ORT_ENABLE_EXTENDED);
-
- opts.EnableCpuMemArena();
allocator_info = Ort::MemoryInfo::CreateCpu(OrtArenaAllocator, OrtMemTypeDefault);
localizer.emplace(
allocator_info,
Ort::Session{env, convert(localizer_model_path_enc).c_str(), opts});
-
+
poseestimator.emplace(
allocator_info,
Ort::Session{env, convert(poseestimator_model_path_enc).c_str(), opts});