chore: include path of onnxruntime>=1.16

horror-proton · May 25, 2024 · 6de0794 · 6de0794
1 parent b5d1b88
commit 6de0794
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/src/MaaCore/Config/OnnxSessions.h b/src/MaaCore/Config/OnnxSessions.h
@@ -4,7 +4,11 @@
 
 #include <unordered_map>
 
+#if __has_include(<onnxruntime_cxx_api.h>)
+#include <onnxruntime_cxx_api.h>
+#else
 #include <onnxruntime/core/session/onnxruntime_cxx_api.h>
+#endif
 
 namespace asst
 {

diff --git a/src/MaaCore/Vision/Battle/BattlefieldClassifier.cpp b/src/MaaCore/Vision/Battle/BattlefieldClassifier.cpp
@@ -6,7 +6,6 @@
 #include <array>
 #include <cmath>
 
-#include <onnxruntime/core/session/onnxruntime_cxx_api.h>
 
 #include "Config/OnnxSessions.h"
 #include "Config/TaskData.h"

diff --git a/src/MaaCore/Vision/Battle/BattlefieldDetector.cpp b/src/MaaCore/Vision/Battle/BattlefieldDetector.cpp
@@ -5,7 +5,6 @@
 #include <cmath>
 
 #include "Utils/NoWarningCV.h"
-#include <onnxruntime/core/session/onnxruntime_cxx_api.h>
 
 #include "Config/Miscellaneous/TilePack.h"
 #include "Config/OnnxSessions.h"