From 5bc5340de6d621e559e916b45c3570533ea9f778 Mon Sep 17 00:00:00 2001
From: Stanislaw Halik <sthalik@misaki.pl>
Date: Wed, 26 Oct 2022 11:05:17 +0200
Subject: tracker/nn: fix build

Issue: #1559
---
 tracker-neuralnet/CMakeLists.txt             | 4 ++++
 tracker-neuralnet/ftnoir_tracker_neuralnet.h | 2 +-
 2 files changed, 5 insertions(+), 1 deletion(-)

(limited to 'tracker-neuralnet')

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>
-- 
cgit v1.2.3