CMakeLists.txt 2.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748
  1. # Openvino
  2. include_directories("${OPENVINO_DIR}")
  3. include_directories("${OPENVINO_DIR}/include")
  4. include_directories("${OPENVINO_DIR}/external/tbb/include/tbb")
  5. link_directories("${OPENVINO_DIR}/lib")
  6. link_directories("${OPENVINO_DIR}/external/tbb/lib")
  7. link_directories("${NGRAPH_LIB}/lib")
  8. include_directories("${GFLAGS_DIR}/include")
  9. link_directories("${GFLAGS_DIR}/lib")
  10. if(WIN32)
  11. link_directories("${OPENVINO_DIR}/lib/intel64/Release")
  12. link_directories("${OPENVINO_DIR}/bin/intel64/Release")
  13. set(DEPS ${DEPS} ${OPENVINO_DIR}/lib/intel64/Release/inference_engine${CMAKE_STATIC_LIBRARY_SUFFIX})
  14. set(DEPS ${DEPS} ${OPENVINO_DIR}/lib/intel64/Release/inference_engine_legacy${CMAKE_STATIC_LIBRARY_SUFFIX})
  15. set(DEPS ${DEPS} gflags_static libyaml-cppmt shlwapi)
  16. else()
  17. if (${ARCH} STREQUAL "armv7")
  18. if(WITH_STATIC_LIB)
  19. set(DEPS ${DEPS} ${OPENVINO_DIR}/lib/armv7l/libinference_engine${CMAKE_STATIC_LIBRARY_SUFFIX})
  20. set(DEPS ${DEPS} ${OPENVINO_DIR}/lib/armv7l/libinference_engine_legacy${CMAKE_STATIC_LIBRARY_SUFFIX})
  21. else()
  22. set(DEPS ${DEPS} ${OPENVINO_DIR}/lib/armv7l/libinference_engine${CMAKE_SHARED_LIBRARY_SUFFIX})
  23. set(DEPS ${DEPS} ${OPENVINO_DIR}/lib/armv7l/libinference_engine_legacy${CMAKE_SHARED_LIBRARY_SUFFIX})
  24. endif()
  25. else()
  26. if(WITH_STATIC_LIB)
  27. set(DEPS ${DEPS} ${OPENVINO_DIR}/lib/intel64/libinference_engine${CMAKE_STATIC_LIBRARY_SUFFIX})
  28. set(DEPS ${DEPS} ${OPENVINO_DIR}/lib/intel64/libinference_engine_legacy${CMAKE_STATIC_LIBRARY_SUFFIX})
  29. else()
  30. set(DEPS ${DEPS} ${OPENVINO_DIR}/lib/intel64/libinference_engine${CMAKE_SHARED_LIBRARY_SUFFIX})
  31. set(DEPS ${DEPS} ${OPENVINO_DIR}/lib/intel64/libinference_engine_legacy${CMAKE_SHARED_LIBRARY_SUFFIX})
  32. endif()
  33. endif()
  34. set(DEPS ${DEPS} gflags yaml-cpp)
  35. endif(WIN32)
  36. message("-----DEPS = ${DEPS}")
  37. # engine src
  38. set(ENGINE_SRC ${PROJECT_SOURCE_DIR}/model_deploy/engine/src/openvino_engine.cpp)
  39. add_executable(model_infer model_infer.cpp ${SRC} ${ENGINE_SRC} ${DETECTOR_SRC})
  40. ADD_DEPENDENCIES(model_infer ext-yaml-cpp)
  41. target_link_libraries(model_infer ${DEPS})