mirror of https://github.com/opencv/opencv.git
Open Source Computer Vision Library
https://opencv.org/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
61 lines
2.0 KiB
61 lines
2.0 KiB
ocv_clear_vars(HAVE_ONNX) |
|
|
|
set(ONNXRT_ROOT_DIR "" CACHE PATH "ONNX Runtime install directory") |
|
|
|
# For now, check the old name ORT_INSTALL_DIR |
|
if(ORT_INSTALL_DIR AND NOT ONNXRT_ROOT_DIR) |
|
set(ONNXRT_ROOT_DIR ${ORT_INSTALL_DIR}) |
|
endif() |
|
|
|
if(ONNXRT_ROOT_DIR) |
|
find_library(ORT_LIB onnxruntime |
|
${ONNXRT_ROOT_DIR}/lib |
|
CMAKE_FIND_ROOT_PATH_BOTH) |
|
# The location of headers varies across different versions of ONNX Runtime |
|
find_path(ORT_INCLUDE onnxruntime_cxx_api.h |
|
${ONNXRT_ROOT_DIR}/include/onnxruntime/ |
|
${ONNXRT_ROOT_DIR}/include/onnxruntime/core/session |
|
CMAKE_FIND_ROOT_PATH_BOTH) |
|
endif() |
|
|
|
macro(detect_onxxrt_ep filename dir have_ep_var) |
|
find_path(ORT_EP_INCLUDE ${filename} ${dir} CMAKE_FIND_ROOT_PATH_BOTH) |
|
if(ORT_EP_INCLUDE) |
|
set(${have_ep_var} TRUE) |
|
endif() |
|
endmacro() |
|
|
|
if(ORT_LIB AND ORT_INCLUDE) |
|
# Check DirectML Execution Provider availability |
|
get_filename_component(dml_dir ${ONNXRT_ROOT_DIR}/include/onnxruntime/core/providers/dml ABSOLUTE) |
|
detect_onxxrt_ep( |
|
dml_provider_factory.h |
|
${dml_dir} |
|
HAVE_ONNX_DML |
|
) |
|
|
|
# Check CoreML Execution Provider availability |
|
get_filename_component(coreml_dir ${ONNXRT_ROOT_DIR}/include/onnxruntime/core/providers/coreml ABSOLUTE) |
|
detect_onxxrt_ep( |
|
coreml_provider_factory.h |
|
${coreml_dir} |
|
HAVE_ONNX_COREML |
|
) |
|
|
|
set(HAVE_ONNX TRUE) |
|
# For CMake output only |
|
set(ONNX_LIBRARIES "${ORT_LIB}" CACHE STRING "ONNX Runtime libraries") |
|
set(ONNX_INCLUDE_DIR "${ORT_INCLUDE}" CACHE STRING "ONNX Runtime include path") |
|
|
|
# Link target with associated interface headers |
|
set(ONNX_LIBRARY "onnxruntime" CACHE STRING "ONNX Link Target") |
|
ocv_add_library(${ONNX_LIBRARY} SHARED IMPORTED) |
|
set_target_properties(${ONNX_LIBRARY} PROPERTIES |
|
INTERFACE_INCLUDE_DIRECTORIES ${ORT_INCLUDE} |
|
IMPORTED_LOCATION ${ORT_LIB} |
|
IMPORTED_IMPLIB ${ORT_LIB}) |
|
endif() |
|
|
|
if(NOT HAVE_ONNX) |
|
ocv_clear_vars(HAVE_ONNX ORT_LIB ORT_INCLUDE_DIR) |
|
endif()
|
|
|