diff options
author | Michael Welter <michael@welter-4d.de> | 2021-10-23 20:52:35 +0200 |
---|---|---|
committer | Michael Welter <michael@welter-4d.de> | 2021-10-23 20:55:20 +0200 |
commit | 6b89c27a5eb681b4403969dce898b74947071a99 (patch) | |
tree | 806788f669e51090e47e22de9ef483ed0deabefa /tracker-neuralnet | |
parent | b3880a052ef103fa9d171158c1ea38a51c220cbd (diff) |
tracker/nn: Remove onnx session options which hurt performance on Linux
Diffstat (limited to 'tracker-neuralnet')
-rw-r--r-- | tracker-neuralnet/ftnoir_tracker_neuralnet.cpp | 6 |
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}); |