summaryrefslogtreecommitdiffhomepage
path: root/tracker-neuralnet
diff options
context:
space:
mode:
Diffstat (limited to 'tracker-neuralnet')
-rw-r--r--tracker-neuralnet/CMakeLists.txt9
-rw-r--r--tracker-neuralnet/ftnoir_tracker_neuralnet.h2
2 files changed, 3 insertions, 8 deletions
diff --git a/tracker-neuralnet/CMakeLists.txt b/tracker-neuralnet/CMakeLists.txt
index 36395684..bcdbbc81 100644
--- a/tracker-neuralnet/CMakeLists.txt
+++ b/tracker-neuralnet/CMakeLists.txt
@@ -5,14 +5,9 @@ find_package(ONNXRuntime QUIET)
if(OpenCV_FOUND AND ONNXRuntime_FOUND AND OpenMP_FOUND)
otr_module(tracker-neuralnet)
- target_include_directories(${self}
- SYSTEM PRIVATE
- ${OpenCV_INCLUDE_DIRS}
- ${ONNXRuntime_INCLUDE_DIRS}
- )
target_link_libraries(${self}
- opentrack-cv
- ${ONNXRuntime_LIBRARIES}
+ opentrack-cv
+ onnxruntime
${OpenCV_LIBS}
OpenMP::OpenMP_C
)
diff --git a/tracker-neuralnet/ftnoir_tracker_neuralnet.h b/tracker-neuralnet/ftnoir_tracker_neuralnet.h
index 6c3af079..7f496dc6 100644
--- a/tracker-neuralnet/ftnoir_tracker_neuralnet.h
+++ b/tracker-neuralnet/ftnoir_tracker_neuralnet.h
@@ -26,7 +26,7 @@
#include <memory>
#include <cinttypes>
-#include <onnxruntime/core/session/onnxruntime_cxx_api.h>
+#include <onnxruntime_cxx_api.h>
#include <opencv2/core.hpp>
#include <opencv2/core/types.hpp>