summaryrefslogtreecommitdiffhomepage
path: root/tracker-neuralnet
diff options
context:
space:
mode:
authorStanislaw Halik <sthalik@misaki.pl>2022-10-26 11:05:17 +0200
committerStanislaw Halik <sthalik@misaki.pl>2022-10-26 11:05:17 +0200
commit5bc5340de6d621e559e916b45c3570533ea9f778 (patch)
treecbed4779be0d00e437bbb0441f08cc9181819340 /tracker-neuralnet
parentc7a94e51b2e3d8162232aaef3d0d9f14330ea165 (diff)
tracker/nn: fix build
Issue: #1559
Diffstat (limited to 'tracker-neuralnet')
-rw-r--r--tracker-neuralnet/CMakeLists.txt4
-rw-r--r--tracker-neuralnet/ftnoir_tracker_neuralnet.h2
2 files changed, 5 insertions, 1 deletions
diff --git a/tracker-neuralnet/CMakeLists.txt b/tracker-neuralnet/CMakeLists.txt
index f414c920..c9d380ab 100644
--- a/tracker-neuralnet/CMakeLists.txt
+++ b/tracker-neuralnet/CMakeLists.txt
@@ -9,6 +9,10 @@ if(OpenCV_FOUND AND ONNXRuntime_FOUND AND OpenMP_FOUND)
add_definitions(-D_HAS_EXCEPTIONS=1)
endif()
+ if(EXISTS "${ONNXRuntime_INCLUDE_DIR}/onnxruntime/core/session/.")
+ include_directories("${ONNXRuntime_INCLUDE_DIR}/onnxruntime/core/session")
+ endif()
+
otr_module(tracker-neuralnet)
target_link_libraries(${self}
diff --git a/tracker-neuralnet/ftnoir_tracker_neuralnet.h b/tracker-neuralnet/ftnoir_tracker_neuralnet.h
index 3ca607c2..9b481186 100644
--- a/tracker-neuralnet/ftnoir_tracker_neuralnet.h
+++ b/tracker-neuralnet/ftnoir_tracker_neuralnet.h
@@ -27,7 +27,7 @@
#include <cinttypes>
#include <array>
-#include <onnxruntime/core/session/onnxruntime_cxx_api.h>
+#include <onnxruntime_cxx_api.h>
#include <opencv2/core.hpp>
#include <opencv2/core/types.hpp>