CMakeLists.txt 12 KB

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