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.
72 lines
2.2 KiB
72 lines
2.2 KiB
# The script detects Intel(R) Inference Engine installation |
|
# |
|
# Parameters: |
|
# INTEL_CVSDK_DIR - Path to Inference Engine root folder |
|
# IE_PLUGINS_PATH - Path to folder with Inference Engine plugins |
|
# |
|
# On return this will define: |
|
# |
|
# HAVE_INF_ENGINE - True if Intel Inference Engine was found |
|
# INF_ENGINE_INCLUDE_DIRS - Inference Engine include folder |
|
# INF_ENGINE_LIBRARIES - Inference Engine libraries and it's dependencies |
|
# |
|
macro(ie_fail) |
|
set(HAVE_INF_ENGINE FALSE) |
|
return() |
|
endmacro() |
|
|
|
|
|
if(NOT INF_ENGINE_ROOT_DIR OR NOT EXISTS "${INF_ENGINE_ROOT_DIR}/include/inference_engine.hpp") |
|
set(ie_root_paths "${INF_ENGINE_ROOT_DIR}") |
|
if(DEFINED ENV{INTEL_CVSDK_DIR}) |
|
list(APPEND ie_root_paths "$ENV{INTEL_CVSDK_DIR}") |
|
list(APPEND ie_root_paths "$ENV{INTEL_CVSDK_DIR}/inference_engine") |
|
endif() |
|
if(DEFINED INTEL_CVSDK_DIR) |
|
list(APPEND ie_root_paths "${INTEL_CVSDK_DIR}") |
|
list(APPEND ie_root_paths "${INTEL_CVSDK_DIR}/inference_engine") |
|
endif() |
|
|
|
if(NOT ie_root_paths) |
|
list(APPEND ie_root_paths "/opt/intel/deeplearning_deploymenttoolkit/deployment_tools/inference_engine") |
|
endif() |
|
|
|
find_path(INF_ENGINE_ROOT_DIR include/inference_engine.hpp PATHS ${ie_root_paths}) |
|
endif() |
|
|
|
set(INF_ENGINE_INCLUDE_DIRS "${INF_ENGINE_ROOT_DIR}/include" CACHE PATH "Path to Inference Engine include directory") |
|
|
|
if(NOT INF_ENGINE_ROOT_DIR |
|
OR NOT EXISTS "${INF_ENGINE_ROOT_DIR}" |
|
OR NOT EXISTS "${INF_ENGINE_INCLUDE_DIRS}" |
|
OR NOT EXISTS "${INF_ENGINE_INCLUDE_DIRS}/inference_engine.hpp" |
|
) |
|
ie_fail() |
|
endif() |
|
|
|
set(INF_ENGINE_LIBRARIES "") |
|
|
|
set(ie_lib_list inference_engine) |
|
|
|
link_directories( |
|
${INTEL_CVSDK_DIR}/external/mklml_lnx/lib |
|
${INTEL_CVSDK_DIR}/inference_engine/external/mklml_lnx/lib |
|
${INTEL_CVSDK_DIR}/inference_engine/external/mkltiny_lnx/lib |
|
${INTEL_CVSDK_DIR}/external/cldnn/lib |
|
${INTEL_CVSDK_DIR}/inference_engine/external/cldnn/lib |
|
) |
|
|
|
foreach(lib ${ie_lib_list}) |
|
find_library(${lib} |
|
NAMES ${lib} |
|
# For inference_engine |
|
HINTS ${IE_PLUGINS_PATH} |
|
HINTS "$ENV{IE_PLUGINS_PATH}" |
|
) |
|
if(NOT ${lib}) |
|
ie_fail() |
|
endif() |
|
list(APPEND INF_ENGINE_LIBRARIES ${${lib}}) |
|
endforeach() |
|
|
|
set(HAVE_INF_ENGINE TRUE)
|
|
|