CMakeLists.txt 13 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301
  1. cmake_minimum_required(VERSION 3.0)
  2. project(PaddleX CXX C)
  3. option(WITH_MKL "Compile demo with MKL/OpenBlas support,defaultuseMKL." ON)
  4. option(WITH_GPU "Compile demo with GPU/CPU, default use CPU." ON)
  5. option(WITH_STATIC_LIB "Compile demo with static/shared library, default use static." OFF)
  6. option(WITH_TENSORRT "Compile demo with TensorRT." OFF)
  7. option(WITH_ENCRYPTION "Compile demo with encryption tool." OFF)
  8. SET(TENSORRT_DIR "" CACHE PATH "Location of libraries")
  9. SET(PADDLE_DIR "" CACHE PATH "Location of libraries")
  10. SET(OPENCV_DIR "" CACHE PATH "Location of libraries")
  11. SET(ENCRYPTION_DIR"" CACHE PATH "Location of libraries")
  12. SET(CUDA_LIB "" CACHE PATH "Location of libraries")
  13. if (NOT WIN32)
  14. set(CMAKE_ARCHIVE_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR}/lib)
  15. set(CMAKE_LIBRARY_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR}/lib)
  16. set(CMAKE_RUNTIME_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR}/demo)
  17. else()
  18. set(CMAKE_ARCHIVE_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR}/paddlex_inference)
  19. set(CMAKE_LIBRARY_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR}/paddlex_inference)
  20. endif()
  21. if (NOT WIN32)
  22. SET(YAML_BUILD_TYPE ON CACHE BOOL "yaml build shared library.")
  23. else()
  24. SET(YAML_BUILD_TYPE OFF CACHE BOOL "yaml build shared library.")
  25. endif()
  26. include(cmake/yaml-cpp.cmake)
  27. include_directories("${CMAKE_SOURCE_DIR}/")
  28. include_directories("${CMAKE_CURRENT_BINARY_DIR}/ext/yaml-cpp/src/ext-yaml-cpp/include")
  29. link_directories("${CMAKE_CURRENT_BINARY_DIR}/ext/yaml-cpp/lib")
  30. macro(safe_set_static_flag)
  31. foreach(flag_var
  32. CMAKE_CXX_FLAGS CMAKE_CXX_FLAGS_DEBUG CMAKE_CXX_FLAGS_RELEASE
  33. CMAKE_CXX_FLAGS_MINSIZEREL CMAKE_CXX_FLAGS_RELWITHDEBINFO)
  34. if(${flag_var} MATCHES "/MD")
  35. string(REGEX REPLACE "/MD" "/MT" ${flag_var} "${${flag_var}}")
  36. endif(${flag_var} MATCHES "/MD")
  37. endforeach(flag_var)
  38. endmacro()
  39. if (WITH_ENCRYPTION)
  40. add_definitions( -DWITH_ENCRYPTION=${WITH_ENCRYPTION})
  41. endif()
  42. if (WITH_MKL)
  43. ADD_DEFINITIONS(-DUSE_MKL)
  44. endif()
  45. if (NOT DEFINED PADDLE_DIR OR ${PADDLE_DIR} STREQUAL "")
  46. message(FATAL_ERROR "please set PADDLE_DIR with -DPADDLE_DIR=/path/paddle_influence_dir")
  47. endif()
  48. if (NOT DEFINED OPENCV_DIR OR ${OPENCV_DIR} STREQUAL "")
  49. message(FATAL_ERROR "please set OPENCV_DIR with -DOPENCV_DIR=/path/opencv")
  50. endif()
  51. include_directories("${CMAKE_SOURCE_DIR}/")
  52. include_directories("${PADDLE_DIR}/")
  53. include_directories("${PADDLE_DIR}/third_party/install/protobuf/include")
  54. include_directories("${PADDLE_DIR}/third_party/install/glog/include")
  55. include_directories("${PADDLE_DIR}/third_party/install/gflags/include")
  56. include_directories("${PADDLE_DIR}/third_party/install/xxhash/include")
  57. if (EXISTS "${PADDLE_DIR}/third_party/install/snappy/include")
  58. include_directories("${PADDLE_DIR}/third_party/install/snappy/include")
  59. endif()
  60. if(EXISTS "${PADDLE_DIR}/third_party/install/snappystream/include")
  61. include_directories("${PADDLE_DIR}/third_party/install/snappystream/include")
  62. endif()
  63. include_directories("${PADDLE_DIR}/third_party/install/zlib/include")
  64. include_directories("${PADDLE_DIR}/third_party/boost")
  65. include_directories("${PADDLE_DIR}/third_party/eigen3")
  66. if (EXISTS "${PADDLE_DIR}/third_party/install/snappy/lib")
  67. link_directories("${PADDLE_DIR}/third_party/install/snappy/lib")
  68. endif()
  69. if(EXISTS "${PADDLE_DIR}/third_party/install/snappystream/lib")
  70. link_directories("${PADDLE_DIR}/third_party/install/snappystream/lib")
  71. endif()
  72. link_directories("${PADDLE_DIR}/third_party/install/zlib/lib")
  73. link_directories("${PADDLE_DIR}/third_party/install/protobuf/lib")
  74. link_directories("${PADDLE_DIR}/third_party/install/glog/lib")
  75. link_directories("${PADDLE_DIR}/third_party/install/gflags/lib")
  76. link_directories("${PADDLE_DIR}/third_party/install/xxhash/lib")
  77. link_directories("${PADDLE_DIR}/paddle/lib/")
  78. link_directories("${CMAKE_CURRENT_BINARY_DIR}")
  79. if (WIN32)
  80. include_directories("${PADDLE_DIR}/paddle/fluid/inference")
  81. include_directories("${PADDLE_DIR}/paddle/include")
  82. link_directories("${PADDLE_DIR}/paddle/fluid/inference")
  83. find_package(OpenCV REQUIRED PATHS ${OPENCV_DIR}/build/ NO_DEFAULT_PATH)
  84. unset(OpenCV_DIR CACHE)
  85. else ()
  86. find_package(OpenCV REQUIRED PATHS ${OPENCV_DIR}/share/OpenCV NO_DEFAULT_PATH)
  87. include_directories("${PADDLE_DIR}/paddle/include")
  88. link_directories("${PADDLE_DIR}/paddle/lib")
  89. endif ()
  90. include_directories(${OpenCV_INCLUDE_DIRS})
  91. if (WIN32)
  92. add_definitions("/DGOOGLE_GLOG_DLL_DECL=")
  93. set(CMAKE_C_FLAGS_DEBUG "${CMAKE_C_FLAGS_DEBUG} /bigobj /MTd")
  94. set(CMAKE_C_FLAGS_RELEASE "${CMAKE_C_FLAGS_RELEASE} /bigobj /MT")
  95. set(CMAKE_CXX_FLAGS_DEBUG "${CMAKE_CXX_FLAGS_DEBUG} /bigobj /MTd")
  96. set(CMAKE_CXX_FLAGS_RELEASE "${CMAKE_CXX_FLAGS_RELEASE} /bigobj /MT")
  97. if (WITH_STATIC_LIB)
  98. safe_set_static_flag()
  99. add_definitions(-DSTATIC_LIB)
  100. endif()
  101. else()
  102. set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -g -o2 -fopenmp -std=c++11")
  103. set(CMAKE_STATIC_LIBRARY_PREFIX "")
  104. endif()
  105. if (WITH_GPU)
  106. if (NOT DEFINED CUDA_LIB OR ${CUDA_LIB} STREQUAL "")
  107. message(FATAL_ERROR "please set CUDA_LIB with -DCUDA_LIB=/path/cuda/lib64")
  108. endif()
  109. if (NOT WIN32)
  110. if (NOT DEFINED CUDNN_LIB)
  111. message(FATAL_ERROR "please set CUDNN_LIB with -DCUDNN_LIB=/path/cudnn/")
  112. endif()
  113. endif(NOT WIN32)
  114. endif()
  115. if (NOT WIN32)
  116. if (WITH_TENSORRT AND WITH_GPU)
  117. include_directories("${TENSORRT_DIR}/include")
  118. link_directories("${TENSORRT_DIR}/lib")
  119. endif()
  120. endif(NOT WIN32)
  121. if (NOT WIN32)
  122. set(NGRAPH_PATH "${PADDLE_DIR}/third_party/install/ngraph")
  123. if(EXISTS ${NGRAPH_PATH})
  124. include(GNUInstallDirs)
  125. include_directories("${NGRAPH_PATH}/include")
  126. link_directories("${NGRAPH_PATH}/${CMAKE_INSTALL_LIBDIR}")
  127. set(NGRAPH_LIB ${NGRAPH_PATH}/${CMAKE_INSTALL_LIBDIR}/libngraph${CMAKE_SHARED_LIBRARY_SUFFIX})
  128. endif()
  129. endif()
  130. if(WITH_MKL)
  131. include_directories("${PADDLE_DIR}/third_party/install/mklml/include")
  132. if (WIN32)
  133. set(MATH_LIB ${PADDLE_DIR}/third_party/install/mklml/lib/mklml.lib
  134. ${PADDLE_DIR}/third_party/install/mklml/lib/libiomp5md.lib)
  135. else ()
  136. set(MATH_LIB ${PADDLE_DIR}/third_party/install/mklml/lib/libmklml_intel${CMAKE_SHARED_LIBRARY_SUFFIX}
  137. ${PADDLE_DIR}/third_party/install/mklml/lib/libiomp5${CMAKE_SHARED_LIBRARY_SUFFIX})
  138. execute_process(COMMAND cp -r ${PADDLE_DIR}/third_party/install/mklml/lib/libmklml_intel${CMAKE_SHARED_LIBRARY_SUFFIX} /usr/lib)
  139. endif ()
  140. set(MKLDNN_PATH "${PADDLE_DIR}/third_party/install/mkldnn")
  141. if(EXISTS ${MKLDNN_PATH})
  142. include_directories("${MKLDNN_PATH}/include")
  143. if (WIN32)
  144. set(MKLDNN_LIB ${MKLDNN_PATH}/lib/mkldnn.lib)
  145. else ()
  146. set(MKLDNN_LIB ${MKLDNN_PATH}/lib/libmkldnn.so.0)
  147. endif ()
  148. endif()
  149. else()
  150. set(MATH_LIB ${PADDLE_DIR}/third_party/install/openblas/lib/libopenblas${CMAKE_STATIC_LIBRARY_SUFFIX})
  151. endif()
  152. if (WIN32)
  153. if(EXISTS "${PADDLE_DIR}/paddle/fluid/inference/libpaddle_fluid${CMAKE_STATIC_LIBRARY_SUFFIX}")
  154. set(DEPS
  155. ${PADDLE_DIR}/paddle/fluid/inference/libpaddle_fluid${CMAKE_STATIC_LIBRARY_SUFFIX})
  156. else()
  157. set(DEPS
  158. ${PADDLE_DIR}/paddle/lib/libpaddle_fluid${CMAKE_STATIC_LIBRARY_SUFFIX})
  159. endif()
  160. endif()
  161. if(WITH_STATIC_LIB)
  162. set(DEPS
  163. ${PADDLE_DIR}/paddle/lib/libpaddle_fluid${CMAKE_STATIC_LIBRARY_SUFFIX})
  164. else()
  165. set(DEPS
  166. ${PADDLE_DIR}/paddle/lib/libpaddle_fluid${CMAKE_SHARED_LIBRARY_SUFFIX})
  167. endif()
  168. if (NOT WIN32)
  169. set(DEPS ${DEPS}
  170. ${MATH_LIB} ${MKLDNN_LIB}
  171. glog gflags protobuf z xxhash yaml-cpp
  172. )
  173. if(EXISTS "${PADDLE_DIR}/third_party/install/snappystream/lib")
  174. set(DEPS ${DEPS} snappystream)
  175. endif()
  176. if (EXISTS "${PADDLE_DIR}/third_party/install/snappy/lib")
  177. set(DEPS ${DEPS} snappy)
  178. endif()
  179. else()
  180. set(DEPS ${DEPS}
  181. ${MATH_LIB} ${MKLDNN_LIB}
  182. glog gflags_static libprotobuf zlibstatic xxhash libyaml-cppmt)
  183. set(DEPS ${DEPS} libcmt shlwapi)
  184. if (EXISTS "${PADDLE_DIR}/third_party/install/snappy/lib")
  185. set(DEPS ${DEPS} snappy)
  186. endif()
  187. if(EXISTS "${PADDLE_DIR}/third_party/install/snappystream/lib")
  188. set(DEPS ${DEPS} snappystream)
  189. endif()
  190. endif(NOT WIN32)
  191. if(WITH_GPU)
  192. if(NOT WIN32)
  193. if (WITH_TENSORRT)
  194. set(DEPS ${DEPS} ${TENSORRT_DIR}/lib/libnvinfer${CMAKE_SHARED_LIBRARY_SUFFIX})
  195. set(DEPS ${DEPS} ${TENSORRT_DIR}/lib/libnvinfer_plugin${CMAKE_SHARED_LIBRARY_SUFFIX})
  196. endif()
  197. set(DEPS ${DEPS} ${CUDA_LIB}/libcudart${CMAKE_SHARED_LIBRARY_SUFFIX})
  198. set(DEPS ${DEPS} ${CUDNN_LIB}/libcudnn${CMAKE_SHARED_LIBRARY_SUFFIX})
  199. else()
  200. set(DEPS ${DEPS} ${CUDA_LIB}/cudart${CMAKE_STATIC_LIBRARY_SUFFIX} )
  201. set(DEPS ${DEPS} ${CUDA_LIB}/cublas${CMAKE_STATIC_LIBRARY_SUFFIX} )
  202. set(DEPS ${DEPS} ${CUDA_LIB}/cudnn${CMAKE_STATIC_LIBRARY_SUFFIX})
  203. endif()
  204. endif()
  205. if(WITH_ENCRYPTION)
  206. if(NOT WIN32)
  207. include_directories("${ENCRYPTION_DIR}/include")
  208. link_directories("${ENCRYPTION_DIR}/lib")
  209. set(DEPS ${DEPS} ${ENCRYPTION_DIR}/lib/libpmodel-decrypt${CMAKE_SHARED_LIBRARY_SUFFIX})
  210. else()
  211. message(FATAL_ERROR "Encryption Tool don't support WINDOWS")
  212. endif()
  213. endif()
  214. if (NOT WIN32)
  215. set(EXTERNAL_LIB "-ldl -lrt -lgomp -lz -lm -lpthread")
  216. set(DEPS ${DEPS} ${EXTERNAL_LIB})
  217. endif()
  218. set(DEPS ${DEPS} ${OpenCV_LIBS})
  219. add_library(paddlex_inference SHARED src/visualize src/transforms.cpp src/paddlex.cpp)
  220. ADD_DEPENDENCIES(paddlex_inference ext-yaml-cpp)
  221. target_link_libraries(paddlex_inference ${DEPS})
  222. add_executable(classifier demo/classifier.cpp src/transforms.cpp src/paddlex.cpp)
  223. ADD_DEPENDENCIES(classifier ext-yaml-cpp)
  224. target_link_libraries(classifier ${DEPS})
  225. add_executable(detector demo/detector.cpp src/transforms.cpp src/paddlex.cpp src/visualize.cpp)
  226. ADD_DEPENDENCIES(detector ext-yaml-cpp)
  227. target_link_libraries(detector ${DEPS})
  228. add_executable(segmenter demo/segmenter.cpp src/transforms.cpp src/paddlex.cpp src/visualize.cpp)
  229. ADD_DEPENDENCIES(segmenter ext-yaml-cpp)
  230. target_link_libraries(segmenter ${DEPS})
  231. if (WIN32 AND WITH_MKL)
  232. add_custom_command(TARGET classifier POST_BUILD
  233. COMMAND ${CMAKE_COMMAND} -E copy_if_different ${PADDLE_DIR}/third_party/install/mklml/lib/mklml.dll ./mklml.dll
  234. COMMAND ${CMAKE_COMMAND} -E copy_if_different ${PADDLE_DIR}/third_party/install/mklml/lib/libiomp5md.dll ./libiomp5md.dll
  235. COMMAND ${CMAKE_COMMAND} -E copy_if_different ${PADDLE_DIR}/third_party/install/mkldnn/lib/mkldnn.dll ./mkldnn.dll
  236. COMMAND ${CMAKE_COMMAND} -E copy_if_different ${PADDLE_DIR}/third_party/install/mklml/lib/mklml.dll ./release/mklml.dll
  237. COMMAND ${CMAKE_COMMAND} -E copy_if_different ${PADDLE_DIR}/third_party/install/mklml/lib/libiomp5md.dll ./release/libiomp5md.dll
  238. COMMAND ${CMAKE_COMMAND} -E copy_if_different ${PADDLE_DIR}/third_party/install/mkldnn/lib/mkldnn.dll ./release/mkldnn.dll
  239. )
  240. add_custom_command(TARGET detector POST_BUILD
  241. COMMAND ${CMAKE_COMMAND} -E copy_if_different ${PADDLE_DIR}/third_party/install/mklml/lib/mklml.dll ./mklml.dll
  242. COMMAND ${CMAKE_COMMAND} -E copy_if_different ${PADDLE_DIR}/third_party/install/mklml/lib/libiomp5md.dll ./libiomp5md.dll
  243. COMMAND ${CMAKE_COMMAND} -E copy_if_different ${PADDLE_DIR}/third_party/install/mkldnn/lib/mkldnn.dll ./mkldnn.dll
  244. COMMAND ${CMAKE_COMMAND} -E copy_if_different ${PADDLE_DIR}/third_party/install/mklml/lib/mklml.dll ./release/mklml.dll
  245. COMMAND ${CMAKE_COMMAND} -E copy_if_different ${PADDLE_DIR}/third_party/install/mklml/lib/libiomp5md.dll ./release/libiomp5md.dll
  246. )
  247. add_custom_command(TARGET segmenter POST_BUILD
  248. COMMAND ${CMAKE_COMMAND} -E copy_if_different ${PADDLE_DIR}/third_party/install/mklml/lib/mklml.dll ./mklml.dll
  249. COMMAND ${CMAKE_COMMAND} -E copy_if_different ${PADDLE_DIR}/third_party/install/mklml/lib/libiomp5md.dll ./libiomp5md.dll
  250. COMMAND ${CMAKE_COMMAND} -E copy_if_different ${PADDLE_DIR}/third_party/install/mkldnn/lib/mkldnn.dll ./mkldnn.dll
  251. COMMAND ${CMAKE_COMMAND} -E copy_if_different ${PADDLE_DIR}/third_party/install/mklml/lib/mklml.dll ./release/mklml.dll
  252. COMMAND ${CMAKE_COMMAND} -E copy_if_different ${PADDLE_DIR}/third_party/install/mklml/lib/libiomp5md.dll ./release/libiomp5md.dll
  253. COMMAND ${CMAKE_COMMAND} -E copy_if_different ${PADDLE_DIR}/third_party/install/mkldnn/lib/mkldnn.dll ./release/mkldnn.dll
  254. )
  255. endif()
  256. file(COPY "${CMAKE_SOURCE_DIR}/include/paddlex/visualize.h"
  257. DESTINATION "${CMAKE_BINARY_DIR}/include/" )
  258. file(COPY "${CMAKE_SOURCE_DIR}/include/paddlex/config_parser.h"
  259. DESTINATION "${CMAKE_BINARY_DIR}/include/" )
  260. file(COPY "${CMAKE_SOURCE_DIR}/include/paddlex/transforms.h"
  261. DESTINATION "${CMAKE_BINARY_DIR}/include/" )
  262. file(COPY "${CMAKE_SOURCE_DIR}/include/paddlex/results.h"
  263. DESTINATION "${CMAKE_BINARY_DIR}/include/" )
  264. file(COPY "${CMAKE_SOURCE_DIR}/include/paddlex/paddlex.h"
  265. DESTINATION "${CMAKE_BINARY_DIR}/include/" )