diff options
| -rw-r--r-- | tracker-neuralnet/CMakeLists.txt | 4 | ||||
| -rw-r--r-- | tracker-neuralnet/ftnoir_tracker_neuralnet.h | 2 |
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> |