vendor: OpenCV 5.0.0 snapshot at 40738fb16ceddb5fb3fea747585f7ce6abb0605b
This commit is contained in:
@@ -0,0 +1,178 @@
|
||||
ocv_clear_vars(HAVE_ONNX ORT_LIB ORT_INCLUDE ONNX_LIBRARIES ONNX_INCLUDE_DIR ONNX_VERSION)
|
||||
|
||||
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(NOT ONNXRT_ROOT_DIR AND DEFINED OpenCV_BINARY_DIR)
|
||||
if(EXISTS "${OpenCV_BINARY_DIR}/3rdparty/onnxruntime")
|
||||
file(GLOB __ort_candidates LIST_DIRECTORIES true
|
||||
"${OpenCV_BINARY_DIR}/3rdparty/onnxruntime/onnxruntime-*")
|
||||
list(LENGTH __ort_candidates __ort_candidates_len)
|
||||
if(__ort_candidates_len GREATER 0)
|
||||
list(GET __ort_candidates 0 ONNXRT_ROOT_DIR)
|
||||
endif()
|
||||
unset(__ort_candidates)
|
||||
unset(__ort_candidates_len)
|
||||
endif()
|
||||
endif()
|
||||
|
||||
set(__ort_hint_roots "")
|
||||
if(ONNXRT_ROOT_DIR)
|
||||
list(APPEND __ort_hint_roots "${ONNXRT_ROOT_DIR}")
|
||||
endif()
|
||||
|
||||
# Prefer CMake config packages if present i.e system-installed ORT
|
||||
find_package(onnxruntime CONFIG QUIET)
|
||||
find_package(ONNXRuntime CONFIG QUIET)
|
||||
|
||||
set(__ort_target "")
|
||||
foreach(t
|
||||
onnxruntime::onnxruntime
|
||||
ONNXRuntime::onnxruntime
|
||||
ONNXRuntime::onnxruntime_shared
|
||||
)
|
||||
if(TARGET ${t})
|
||||
set(__ort_target "${t}")
|
||||
break()
|
||||
endif()
|
||||
endforeach()
|
||||
|
||||
if(__ort_target)
|
||||
get_target_property(ORT_INCLUDE ${__ort_target} INTERFACE_INCLUDE_DIRECTORIES)
|
||||
if(ORT_INCLUDE AND NOT IS_DIRECTORY "${ORT_INCLUDE}")
|
||||
list(GET ORT_INCLUDE 0 ORT_INCLUDE)
|
||||
endif()
|
||||
get_target_property(ORT_LIB ${__ort_target} IMPORTED_LOCATION)
|
||||
if(NOT ORT_LIB)
|
||||
get_target_property(ORT_LIB ${__ort_target} IMPORTED_LOCATION_RELEASE)
|
||||
endif()
|
||||
if(NOT ORT_LIB)
|
||||
get_target_property(ORT_LIB ${__ort_target} LOCATION)
|
||||
endif()
|
||||
endif()
|
||||
|
||||
# Locate headers and libraries via find_* in system paths and/or ONNXRT_ROOT_DIR.
|
||||
if(NOT ORT_LIB)
|
||||
find_library(ORT_LIB NAMES onnxruntime
|
||||
HINTS ${__ort_hint_roots}
|
||||
PATH_SUFFIXES lib lib64
|
||||
CMAKE_FIND_ROOT_PATH_BOTH)
|
||||
endif()
|
||||
|
||||
if(NOT ORT_INCLUDE)
|
||||
find_path(ORT_INCLUDE NAMES onnxruntime_cxx_api.h
|
||||
HINTS ${__ort_hint_roots}
|
||||
PATH_SUFFIXES
|
||||
include
|
||||
include/onnxruntime
|
||||
include/onnxruntime/core/session
|
||||
CMAKE_FIND_ROOT_PATH_BOTH)
|
||||
endif()
|
||||
|
||||
unset(__ort_hint_roots)
|
||||
|
||||
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)
|
||||
set(__ort_root_for_ep "${ONNXRT_ROOT_DIR}")
|
||||
if(NOT __ort_root_for_ep)
|
||||
set(__ort_root_for_ep "${ORT_INCLUDE}")
|
||||
string(REGEX REPLACE "(/include/onnxruntime.*)$" "" __ort_root_for_ep "${__ort_root_for_ep}")
|
||||
endif()
|
||||
|
||||
# Check DirectML Execution Provider availability
|
||||
get_filename_component(dml_dir ${__ort_root_for_ep}/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 ${__ort_root_for_ep}/include/onnxruntime/core/providers/coreml ABSOLUTE)
|
||||
detect_onxxrt_ep(
|
||||
coreml_provider_factory.h
|
||||
${coreml_dir}
|
||||
HAVE_ONNX_COREML
|
||||
)
|
||||
|
||||
set(HAVE_ONNX TRUE)
|
||||
# Try to report a human-readable ONNX Runtime version in diagnostics/build info.
|
||||
if(DEFINED onnxruntime_VERSION AND onnxruntime_VERSION)
|
||||
set(ONNX_VERSION "${onnxruntime_VERSION}")
|
||||
elseif(DEFINED ONNXRuntime_VERSION AND ONNXRuntime_VERSION)
|
||||
set(ONNX_VERSION "${ONNXRuntime_VERSION}")
|
||||
elseif(DEFINED ONNXRUNTIME_VERSION AND ONNXRUNTIME_VERSION)
|
||||
set(ONNX_VERSION "${ONNXRUNTIME_VERSION}")
|
||||
else()
|
||||
set(__ort_version_candidates "")
|
||||
if(ONNXRT_ROOT_DIR)
|
||||
list(APPEND __ort_version_candidates "${ONNXRT_ROOT_DIR}")
|
||||
endif()
|
||||
if(ORT_LIB)
|
||||
list(APPEND __ort_version_candidates "${ORT_LIB}")
|
||||
endif()
|
||||
foreach(__ort_version_candidate ${__ort_version_candidates})
|
||||
string(REGEX MATCH "([0-9]+\\.[0-9]+\\.[0-9]+([.-][0-9A-Za-z]+)?)" __ort_version_match "${__ort_version_candidate}")
|
||||
if(__ort_version_match)
|
||||
set(ONNX_VERSION "${__ort_version_match}")
|
||||
break()
|
||||
endif()
|
||||
endforeach()
|
||||
unset(__ort_version_candidate)
|
||||
unset(__ort_version_candidates)
|
||||
unset(__ort_version_match)
|
||||
endif()
|
||||
|
||||
# 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")
|
||||
if(NOT ONNX_VERSION)
|
||||
set(ONNX_VERSION "unknown")
|
||||
endif()
|
||||
set(ONNX_VERSION "${ONNX_VERSION}" CACHE STRING "ONNX Runtime version")
|
||||
|
||||
# Link target with associated interface headers
|
||||
set(ONNX_LIBRARY "onnxruntime" CACHE STRING "ONNX Link Target")
|
||||
if(NOT TARGET ${ONNX_LIBRARY})
|
||||
ocv_add_library(${ONNX_LIBRARY} SHARED IMPORTED)
|
||||
endif()
|
||||
|
||||
if(WIN32)
|
||||
# ORT_LIB is typically the import library (.lib). Prefer matching runtime DLL if available.
|
||||
set(__ort_dll "")
|
||||
if(ONNXRT_ROOT_DIR)
|
||||
find_file(__ort_dll NAMES onnxruntime.dll HINTS "${ONNXRT_ROOT_DIR}" PATH_SUFFIXES bin)
|
||||
endif()
|
||||
if(__ort_dll)
|
||||
set_target_properties(${ONNX_LIBRARY} PROPERTIES
|
||||
INTERFACE_INCLUDE_DIRECTORIES "${ORT_INCLUDE}"
|
||||
IMPORTED_LOCATION "${__ort_dll}"
|
||||
IMPORTED_IMPLIB "${ORT_LIB}")
|
||||
else()
|
||||
set_target_properties(${ONNX_LIBRARY} PROPERTIES
|
||||
INTERFACE_INCLUDE_DIRECTORIES "${ORT_INCLUDE}"
|
||||
IMPORTED_LOCATION "${ORT_LIB}"
|
||||
IMPORTED_IMPLIB "${ORT_LIB}")
|
||||
endif()
|
||||
unset(__ort_dll)
|
||||
else()
|
||||
set_target_properties(${ONNX_LIBRARY} PROPERTIES
|
||||
INTERFACE_INCLUDE_DIRECTORIES "${ORT_INCLUDE}"
|
||||
IMPORTED_LOCATION "${ORT_LIB}")
|
||||
endif()
|
||||
unset(__ort_root_for_ep)
|
||||
endif()
|
||||
|
||||
if(NOT HAVE_ONNX)
|
||||
ocv_clear_vars(HAVE_ONNX ORT_LIB ORT_INCLUDE ONNX_LIBRARIES ONNX_INCLUDE_DIR ONNX_VERSION)
|
||||
endif()
|
||||
Reference in New Issue
Block a user