mirror of https://github.com/opencv/opencv.git
Maksim Shabunin
7 years ago
parent
9433784e47
commit
7cf52de47e
5 changed files with 95 additions and 85 deletions
@ -1,83 +1,87 @@ |
|||||||
# The script detects Intel(R) Inference Engine installation |
# The script detects Intel(R) Inference Engine installation |
||||||
# |
# |
||||||
# Parameters: |
# Cache variables: |
||||||
# INTEL_CVSDK_DIR - Path to Inference Engine root folder |
# INF_ENGINE_OMP_DIR - directory with OpenMP library to link with (needed by some versions of IE) |
||||||
# IE_PLUGINS_PATH - Path to folder with Inference Engine plugins |
# INF_ENGINE_RELEASE - a number reflecting IE source interface (linked with OpenVINO release) |
||||||
# |
# |
||||||
# On return this will define: |
# Detect parameters: |
||||||
|
# 1. Native cmake IE package: |
||||||
|
# - enironment variable InferenceEngine_DIR is set to location of cmake module |
||||||
|
# 2. Custom location: |
||||||
|
# - INF_ENGINE_INCLUDE_DIRS - headers search location |
||||||
|
# - INF_ENGINE_LIB_DIRS - library search location |
||||||
|
# 3. OpenVINO location: |
||||||
|
# - environment variable INTEL_CVSDK_DIR is set to location of OpenVINO installation dir |
||||||
|
# - INF_ENGINE_PLATFORM - part of name of library directory representing its platform (default ubuntu_16.04) |
||||||
# |
# |
||||||
# HAVE_INF_ENGINE - True if Intel Inference Engine was found |
# Result: |
||||||
# INF_ENGINE_INCLUDE_DIRS - Inference Engine include folder |
# INF_ENGINE_TARGET - set to name of imported library target representing InferenceEngine |
||||||
# INF_ENGINE_LIBRARIES - Inference Engine libraries and it's dependencies |
|
||||||
# |
# |
||||||
macro(ie_fail) |
|
||||||
set(HAVE_INF_ENGINE FALSE) |
|
||||||
return() |
|
||||||
endmacro() |
|
||||||
|
|
||||||
if(NOT HAVE_CXX11) |
if(NOT HAVE_CXX11) |
||||||
message(WARNING "DL Inference engine requires C++11. You can turn it on via ENABLE_CXX11=ON CMake flag.") |
message(WARNING "DL Inference engine requires C++11. You can turn it on via ENABLE_CXX11=ON CMake flag.") |
||||||
ie_fail() |
return() |
||||||
endif() |
|
||||||
|
|
||||||
find_package(InferenceEngine QUIET) |
|
||||||
if(InferenceEngine_FOUND) |
|
||||||
set(INF_ENGINE_LIBRARIES "${InferenceEngine_LIBRARIES}") |
|
||||||
set(INF_ENGINE_INCLUDE_DIRS "${InferenceEngine_INCLUDE_DIRS}") |
|
||||||
set(INF_ENGINE_VERSION "${InferenceEngine_VERSION}") |
|
||||||
set(HAVE_INF_ENGINE TRUE) |
|
||||||
return() |
|
||||||
endif() |
endif() |
||||||
|
|
||||||
ocv_check_environment_variables(INTEL_CVSDK_DIR INF_ENGINE_ROOT_DIR IE_PLUGINS_PATH) |
# ======================= |
||||||
|
|
||||||
if(NOT INF_ENGINE_ROOT_DIR OR NOT EXISTS "${INF_ENGINE_ROOT_DIR}/include/inference_engine.hpp") |
function(add_custom_ie_build _inc _lib _lib_rel _lib_dbg _msg) |
||||||
set(ie_root_paths "${INF_ENGINE_ROOT_DIR}") |
if(NOT _inc OR NOT (_lib OR _lib_rel OR _lib_dbg)) |
||||||
if(DEFINED INTEL_CVSDK_DIR) |
return() |
||||||
list(APPEND ie_root_paths "${INTEL_CVSDK_DIR}/") |
endif() |
||||||
list(APPEND ie_root_paths "${INTEL_CVSDK_DIR}/deployment_tools/inference_engine") |
add_library(inference_engine UNKNOWN IMPORTED) |
||||||
endif() |
set_target_properties(inference_engine PROPERTIES |
||||||
|
IMPORTED_LOCATION "${_lib}" |
||||||
|
IMPORTED_IMPLIB_RELEASE "${_lib_rel}" |
||||||
|
IMPORTED_IMPLIB_DEBUG "${_lib_dbg}" |
||||||
|
INTERFACE_INCLUDE_DIRECTORIES "${_inc}" |
||||||
|
) |
||||||
|
find_library(omp_lib iomp5 PATHS "${INF_ENGINE_OMP_DIR}" NO_DEFAULT_PATH) |
||||||
|
if(NOT omp_lib) |
||||||
|
message(WARNING "OpenMP for IE have not been found. Set INF_ENGINE_OMP_DIR variable if you experience build errors.") |
||||||
|
else() |
||||||
|
set_target_properties(inference_engine PROPERTIES IMPORTED_LINK_INTERFACE_LIBRARIES "${omp_lib}") |
||||||
|
endif() |
||||||
|
set(INF_ENGINE_VERSION "Unknown" CACHE STRING "") |
||||||
|
set(INF_ENGINE_TARGET inference_engine PARENT_SCOPE) |
||||||
|
message(STATUS "Detected InferenceEngine: ${_msg}") |
||||||
|
endfunction() |
||||||
|
|
||||||
if(NOT ie_root_paths) |
# ====================== |
||||||
list(APPEND ie_root_paths "/opt/intel/computer_vision_sdk/deployment_tools/inference_engine/") |
|
||||||
endif() |
|
||||||
|
|
||||||
find_path(INF_ENGINE_ROOT_DIR include/inference_engine.hpp PATHS ${ie_root_paths}) |
find_package(InferenceEngine QUIET) |
||||||
if(INF_ENGINE_ROOT_DIR MATCHES "-NOTFOUND$") |
if(InferenceEngine_FOUND) |
||||||
unset(INF_ENGINE_ROOT_DIR CACHE) |
set(INF_ENGINE_TARGET IE::inference_engine) |
||||||
endif() |
set(INF_ENGINE_VERSION "${InferenceEngine_VERSION}" CACHE STRING "") |
||||||
|
message(STATUS "Detected InferenceEngine: cmake package") |
||||||
endif() |
endif() |
||||||
|
|
||||||
set(INF_ENGINE_INCLUDE_DIRS "${INF_ENGINE_ROOT_DIR}/include" CACHE PATH "Path to Inference Engine include directory") |
if(NOT INF_ENGINE_TARGET AND INF_ENGINE_LIB_DIRS AND INF_ENGINE_INCLUDE_DIRS) |
||||||
|
find_path(ie_custom_inc "inference_engine.hpp" PATHS "${INF_ENGINE_INCLUDE_DIRS}" NO_DEFAULT_PATH) |
||||||
if(NOT INF_ENGINE_ROOT_DIR |
find_library(ie_custom_lib "inference_engine" PATHS "${INF_ENGINE_LIB_DIRS}" NO_DEFAULT_PATH) |
||||||
OR NOT EXISTS "${INF_ENGINE_ROOT_DIR}" |
find_library(ie_custom_lib_rel "inference_engine" PATHS "${INF_ENGINE_LIB_DIRS}/Release" NO_DEFAULT_PATH) |
||||||
OR NOT EXISTS "${INF_ENGINE_ROOT_DIR}/include/inference_engine.hpp" |
find_library(ie_custom_lib_dbg "inference_engine" PATHS "${INF_ENGINE_LIB_DIRS}/Debug" NO_DEFAULT_PATH) |
||||||
) |
add_custom_ie_build("${ie_custom_inc}" "${ie_custom_lib}" "${ie_custom_lib_rel}" "${ie_custom_lib_dbg}" "INF_ENGINE_{INCLUDE,LIB}_DIRS") |
||||||
message(WARNING "DL IE: Can't detect INF_ENGINE_ROOT_DIR location.") |
|
||||||
ie_fail() |
|
||||||
endif() |
endif() |
||||||
|
|
||||||
set(INF_ENGINE_LIBRARIES "") |
set(_loc "$ENV{INTEL_CVSDK_DIR}") |
||||||
|
if(NOT INF_ENGINE_TARGET AND _loc) |
||||||
set(ie_lib_list inference_engine) |
set(INF_ENGINE_PLATFORM "ubuntu_16.04" CACHE STRING "InferenceEngine platform (library dir)") |
||||||
|
find_path(ie_custom_env_inc "inference_engine.hpp" PATHS "${_loc}/deployment_tools/inference_engine/include" NO_DEFAULT_PATH) |
||||||
if(NOT IS_ABSOLUTE "${IE_PLUGINS_PATH}") |
find_library(ie_custom_env_lib "inference_engine" PATHS "${_loc}/deployment_tools/inference_engine/lib/${INF_ENGINE_PLATFORM}/intel64" NO_DEFAULT_PATH) |
||||||
set(IE_PLUGINS_PATH "${INF_ENGINE_ROOT_DIR}/${IE_PLUGINS_PATH}") |
find_library(ie_custom_env_lib_rel "inference_engine" PATHS "${_loc}/deployment_tools/inference_engine/lib/intel64/Release" NO_DEFAULT_PATH) |
||||||
|
find_library(ie_custom_env_lib_dbg "inference_engine" PATHS "${_loc}/deployment_tools/inference_engine/lib/intel64/Debug" NO_DEFAULT_PATH) |
||||||
|
add_custom_ie_build("${ie_custom_env_inc}" "${ie_custom_env_lib}" "${ie_custom_env_lib_rel}" "${ie_custom_env_lib_dbg}" "OpenVINO (${_loc})") |
||||||
endif() |
endif() |
||||||
|
|
||||||
link_directories( |
# Add more features to the target |
||||||
${INF_ENGINE_ROOT_DIR}/external/mkltiny_lnx/lib |
|
||||||
${INF_ENGINE_ROOT_DIR}/external/cldnn/lib |
|
||||||
) |
|
||||||
|
|
||||||
foreach(lib ${ie_lib_list}) |
|
||||||
find_library(${lib} NAMES ${lib} HINTS ${IE_PLUGINS_PATH}) |
|
||||||
if(NOT ${lib}) |
|
||||||
message(WARNING "DL IE: Can't find library: '${lib}'") |
|
||||||
ie_fail() |
|
||||||
endif() |
|
||||||
list(APPEND INF_ENGINE_LIBRARIES ${${lib}}) |
|
||||||
endforeach() |
|
||||||
|
|
||||||
set(HAVE_INF_ENGINE TRUE) |
if(INF_ENGINE_TARGET) |
||||||
|
if(NOT INF_ENGINE_RELEASE) |
||||||
|
message(WARNING "InferenceEngine version have not been set, 2018R2 will be used by default. Set INF_ENGINE_RELEASE variable if you experience build errors.") |
||||||
|
endif() |
||||||
|
set(INF_ENGINE_RELEASE "2018020000" CACHE STRING "Force IE version, should be in form YYYYAABBCC (e.g. 2018R2.0.2 -> 2018020002)") |
||||||
|
set_target_properties(${INF_ENGINE_TARGET} PROPERTIES |
||||||
|
INTERFACE_COMPILE_DEFINITIONS "HAVE_INF_ENGINE=1;INF_ENGINE_RELEASE=${INF_ENGINE_RELEASE}" |
||||||
|
) |
||||||
|
endif() |
||||||
|
|||||||
Loading…
Reference in new issue