1 ocv_clear_vars(HAVE_ONNX)
3 set(ONNXRT_ROOT_DIR "" CACHE PATH "ONNX Runtime install directory")
5 # For now, check the old name ORT_INSTALL_DIR
6 if(ORT_INSTALL_DIR AND NOT ONNXRT_ROOT_DIR)
7 set(ONNXRT_ROOT_DIR ${ORT_INSTALL_DIR})
11 find_library(ORT_LIB onnxruntime
12 ${ONNXRT_ROOT_DIR}/lib
13 CMAKE_FIND_ROOT_PATH_BOTH)
14 find_path(ORT_INCLUDE onnxruntime_cxx_api.h
15 ${ONNXRT_ROOT_DIR}/include/onnxruntime/core/session
16 CMAKE_FIND_ROOT_PATH_BOTH)
19 if(ORT_LIB AND ORT_INCLUDE)
21 # For CMake output only
22 set(ONNX_LIBRARIES "${ORT_LIB}" CACHE STRING "ONNX Runtime libraries")
23 set(ONNX_INCLUDE_DIR "${ORT_INCLUDE}" CACHE STRING "ONNX Runtime include path")
25 # Link target with associated interface headers
26 set(ONNX_LIBRARY "onnxruntime" CACHE STRING "ONNX Link Target")
27 ocv_add_library(${ONNX_LIBRARY} SHARED IMPORTED)
28 set_target_properties(${ONNX_LIBRARY} PROPERTIES
29 INTERFACE_INCLUDE_DIRECTORIES ${ORT_INCLUDE}
30 IMPORTED_LOCATION ${ORT_LIB}
31 IMPORTED_IMPLIB ${ORT_LIB})
35 ocv_clear_vars(HAVE_ONNX ORT_LIB ORT_INCLUDE_DIR)