CMakeLists.txt 34.5 KB
Newer Older
B
bikz05 已提交
1 2
### VERSION INFO
set(OpenPose_VERSION_MAJOR 1)
G
gineshidalgo99 已提交
3 4
set(OpenPose_VERSION_MINOR 3)
set(OpenPose_VERSION_PATCH 0)
B
bikz05 已提交
5 6
set(OpenPose_VERSION ${OpenPose_VERSION_MAJOR}.${OpenPose_VERSION_MINOR}.${OpenPose_VERSION_PATCH})

7

8 9 10 11
### OS-DEPENDENT FLAGS
set(CMAKE_MACOSX_RPATH 1)


12
### CMAKE HEADERS
B
Bikramjot Hanzra 已提交
13 14 15 16 17 18
if (${CMAKE_VERSION} VERSION_GREATER 3.0.0)
 cmake_policy(SET CMP0048 NEW)
 project(OpenPose VERSION ${OpenPose_VERSION})
else (${CMAKE_VERSION} VERSION_GREATER 3.0.0)
 project(OpenPose)
endif (${CMAKE_VERSION} VERSION_GREATER 3.0.0)
19 20 21
cmake_minimum_required(VERSION 2.8.7 FATAL_ERROR) # min. cmake version recommended by Caffe


22 23
### Extra functionality
include(cmake/Utils.cmake)
G
gineshidalgo99 已提交
24 25 26
if (NOT WIN32 AND NOT UNIX AND NOT APPLE)
  message(FATAL_ERROR "Unsupported operating system. Only Windows, Mac and Unix systems supported.")
endif (NOT WIN32 AND NOT UNIX AND NOT APPLE)
B
Bikramjot Hanzra 已提交
27 28


G
gineshidalgo99 已提交
29
### FLAGS
30
if (WIN32)
G
gineshidalgo99 已提交
31
  # TODO -- Makeshift solution -- This prevents rerunning build again
32 33 34
  # https://gitlab.kitware.com/cmake/cmake/issues/16783
  set(CMAKE_SUPPRESS_REGENERATION ON)

A
arkceajin 已提交
35 36
  string (REPLACE "/D_WINDOWS" "" CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}")
  string (REPLACE "/DWIN32" "" CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}")
37 38 39 40 41

  # /GL option causes the code to crash -- fix this
  # sdl flags causes error -- error : unknown attribute \"guard\"

  set(CMAKE_CONFIGURATION_TYPES Debug Release CACHE TYPE INTERNAL FORCE)
42

43 44 45 46 47 48 49 50 51
  set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} /MP")
  string(REPLACE "/W3" "/W4" CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}")
  string(REPLACE "/GR" "" CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}")

  string(REPLACE "/O2" "/Ox" CMAKE_CXX_FLAGS_RELEASE "${CMAKE_CXX_FLAGS_RELEASE}")
  string(REPLACE "/Ob2" "" CMAKE_CXX_FLAGS_RELEASE "${CMAKE_CXX_FLAGS_RELEASE}")
  set(CMAKE_CXX_FLAGS_RELEASE "${CMAKE_CXX_FLAGS_RELEASE} /Ot /Oi /Gy /Z7")

  set(CMAKE_SHARED_LINKER_FLAGS_RELEASE "${CMAKE_SHARED_LINKER_FLAGS_RELEASE} /LTCG:incremental /OPT:REF /OPT:ICF")
52

53 54 55
  string(REPLACE "/MDd" "/MD" CMAKE_CXX_FLAGS_DEBUG "${CMAKE_CXX_FLAGS_DEBUG}")
  string(REPLACE "/Zi" "/Z7" CMAKE_CXX_FLAGS_DEBUG "${CMAKE_CXX_FLAGS_DEBUG}")
  string(REPLACE "/RTC1" "" CMAKE_CXX_FLAGS_DEBUG "${CMAKE_CXX_FLAGS_DEBUG}")
56
elseif (UNIX)
G
gineshidalgo99 已提交
57
  # Turn on C++11
58 59
  add_definitions(-std=c++11)
  set(CMAKE_CXX_FLAGS_RELEASE "-O3")
60 61 62 63
elseif (APPLE)
  # Turn on C++11
  set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11")
  set(CMAKE_CXX_FLAGS_RELEASE "-O3")
64
endif (WIN32)
B
Bikramjot Hanzra 已提交
65 66

# C++ additional flags
G
gineshidalgo99 已提交
67
if (CMAKE_COMPILER_IS_GNUCXX)
68 69 70 71 72 73 74 75

  # Select the Enhanced Instruction Set
  set(INSTRUCTION_SET NONE CACHE STRING "Enable Enhanced Instruction Set")
  set_property(CACHE INSTRUCTION_SET PROPERTY STRINGS NONE SSE4 AVX)

  if (${INSTRUCTION_SET} MATCHES "SSE4")
    add_definitions("-DWITH_SSE4")
    set(SIMD_FLAGS "${SIMD_FLAGS} -msse4.1")
76
  endif (${INSTRUCTION_SET} MATCHES "SSE4")
77 78 79 80

  if (${INSTRUCTION_SET} MATCHES "AVX")
    add_definitions("-DWITH_AVX")
    set(SIMD_FLAGS "${SIMD_FLAGS} -mavx")
81
  endif (${INSTRUCTION_SET} MATCHES "AVX")
82

G
gineshidalgo99 已提交
83
  message(STATUS "GCC detected, adding compile flags")
84
  set(OP_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${SIMD_FLAGS} -fopenmp -Wpedantic -Wall -Wextra -Wfatal-errors")
G
gineshidalgo99 已提交
85
endif (CMAKE_COMPILER_IS_GNUCXX)
B
Bikramjot Hanzra 已提交
86 87


G
gineshidalgo99 已提交
88
### PROJECT OPTIONS
B
Bikramjot Hanzra 已提交
89 90 91 92 93 94 95 96 97
# Select the DL Framework
set(DL_FRAMEWORK CAFFE CACHE STRING "Select Deep Learning Framework.")
set_property(CACHE DL_FRAMEWORK PROPERTY STRINGS CAFFE)
# set_property(CACHE DL_FRAMEWORK PROPERTY STRINGS CAFFE CAFFE2 TENSORFLOW)

# Suboptions for Caffe DL Framework
include(CMakeDependentOption)
if (${DL_FRAMEWORK} MATCHES "CAFFE")
  CMAKE_DEPENDENT_OPTION(BUILD_CAFFE "Build Caffe as part of OpenPose." ON
G
gineshidalgo99 已提交
98
    "DL_FRAMEWORK" ON)
99 100
  # OpenPose flags
  add_definitions(-DUSE_CAFFE)
B
Bikramjot Hanzra 已提交
101 102
endif (${DL_FRAMEWORK} MATCHES "CAFFE")

103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121
if (WIN32)
  # Select the Enhanced Instruction Set
  set(INSTRUCTION_SET NONE CACHE STRING "Enable Enhanced Instruction Set")
  set_property(CACHE INSTRUCTION_SET PROPERTY STRINGS NONE SSE SSE2 AVX AVX2 IA32)

  # Suboptions for Enhanced Instruction Set
  if (${INSTRUCTION_SET} MATCHES "SSE")
    add_definitions("/arch:SSE")
  elseif (${INSTRUCTION_SET} MATCHES "SSE2")
    add_definitions("/arch:SSE2")
  elseif (${INSTRUCTION_SET} MATCHES "AVX")
    add_definitions("/arch:AVX")
  elseif (${INSTRUCTION_SET} MATCHES "AVX2")
    add_definitions("/arch:AVX2")
  elseif (${INSTRUCTION_SET} MATCHES "IA32")
    add_definitions("/arch:IA32")
  endif (${INSTRUCTION_SET} MATCHES "SSE")
endif (WIN32)

B
Bikramjot Hanzra 已提交
122
# Set the acceleration library
G
gineshidalgo99 已提交
123 124 125 126 127
if (APPLE)
  set(GPU_MODE CPU_ONLY CACHE STRING "Select the acceleration GPU library or CPU otherwise.")
else (APPLE)
  set(GPU_MODE CUDA CACHE STRING "Select the acceleration GPU library or CPU otherwise.")
endif (APPLE)
128 129 130 131
# Display proper options to user
# if (CUDA_FOUND AND OpenCL_FOUND)
#   set_property(CACHE GPU_MODE PROPERTY STRINGS CUDA OPENCL CPU_ONLY)
# elseif (CUDA_FOUND)
G
gineshidalgo99 已提交
132
#   set_property(CACHE GPU_MODE PROPERTY STRINGS CUDA CPU_ONLY)
133 134 135 136
# elseif (OpenCL_FOUND)
#   set_property(CACHE GPU_MODE PROPERTY STRINGS OPENCL CPU_ONLY)
# else ()
#   set_property(CACHE GPU_MODE PROPERTY STRINGS CPU_ONLY)
137
# endif (CUDA_FOUND AND OpenCL_FOUND)
R
Raaj 已提交
138
set_property(CACHE GPU_MODE PROPERTY STRINGS CUDA OPENCL CPU_ONLY)
G
gineshidalgo99 已提交
139

140
# Look for CUDA
G
gineshidalgo99 已提交
141
set(CUDA_FOUND FALSE)
142 143 144 145 146
if (${GPU_MODE} MATCHES "CUDA")
  find_package(CUDA)
endif (${GPU_MODE} MATCHES "CUDA")
# Look for OpenCL
set(OpenCL_FOUND FALSE)
G
gineshidalgo99 已提交
147
set(CUDA_VERSION_MAJOR 0)
148 149 150 151
if (${GPU_MODE} MATCHES "OPENCL")
  find_package(OpenCL)
endif (${GPU_MODE} MATCHES "OPENCL")

152 153 154 155 156 157 158
# Code to avoid crash at compiling time if OpenCL is not found
if (NOT OpenCL_FOUND)
  set(OpenCL_LIBRARIES "")
endif (NOT OpenCL_FOUND)
# Required for OpenCL in Nvidia graphic cards
if (CUDA_FOUND AND OpenCL_FOUND AND ${CUDA_VERSION_MAJOR} LESS 9)
  add_definitions(-DLOWER_CL_VERSION)
159
endif (CUDA_FOUND AND OpenCL_FOUND AND ${CUDA_VERSION_MAJOR} LESS 9)
160
# Handle desired GPU mode option
161 162 163
if (${GPU_MODE} MATCHES "CUDA")
  # OpenPose flags
  add_definitions(-DUSE_CUDA)
G
gineshidalgo99 已提交
164
  message(STATUS "Building with CUDA.")
165
elseif (${GPU_MODE} MATCHES "CPU_ONLY")
G
gineshidalgo99 已提交
166 167
  add_definitions(-DUSE_CPU_ONLY)
  message(STATUS "Building CPU Only.")
168 169
  # OpenPose flag for Caffe
  add_definitions(-DCPU_ONLY)
170 171 172 173 174
elseif (${GPU_MODE} MATCHES "OPENCL")
  # OpenPose flag for Caffe
  add_definitions(-DUSE_OPENCL)
  add_definitions(-DUSE_GREENTEA)
  message(STATUS "Building with OpenCL.")
175
endif (${GPU_MODE} MATCHES "CUDA")
G
gineshidalgo99 已提交
176 177 178 179 180

# Intel branch with MKL Support
if (${GPU_MODE} MATCHES "CPU_ONLY")
  if (UNIX AND NOT APPLE)
    OPTION(USE_MKL "Use MKL Intel Branch." ON)
181 182
  endif (UNIX AND NOT APPLE)
endif (${GPU_MODE} MATCHES "CPU_ONLY")
B
Bikramjot Hanzra 已提交
183

184 185 186 187 188 189
if (${USE_MKL})
  # OpenPose flags
  add_definitions(-DUSE_MKL)
  message(STATUS "Building with MKL support.")
endif (${USE_MKL})

190 191 192 193 194
# Set/disable profiler
if (PROFILER_ENABLED)
  add_definitions(-DPROFILER_ENABLED)
endif (PROFILER_ENABLED)

B
Bikramjot Hanzra 已提交
195 196 197 198 199 200 201 202 203 204
# Suboptions for GPU architectures
if (${GPU_MODE} MATCHES "CUDA")
  set(CUDA_ARCH Auto CACHE STRING "Select target NVIDIA GPU achitecture.")
  set_property(CACHE CUDA_ARCH PROPERTY STRINGS Auto All Manual)
endif (${GPU_MODE} MATCHES "CUDA")

# Suboptions for acceleration library
if (${GPU_MODE} MATCHES "CUDA")
  option(USE_CUDNN "Build OpenPose with cuDNN library support." ON)
endif (${GPU_MODE} MATCHES "CUDA")
205

G
gineshidalgo99 已提交
206
# Suboptions for OpenPose 3D Reconstruction module and demo
207
option(WITH_3D_RENDERER "Add OpenPose 3D renderer module (it requires FreeGLUT library)." OFF)
208
if (UNIX AND NOT APPLE)
G
gineshidalgo99 已提交
209
  option(WITH_CERES "Add Ceres support for advanced 3-D reconstruction." OFF)
210
endif (UNIX AND NOT APPLE)
211
option(WITH_FLIR_CAMERA "Add FLIR (formerly Point Grey) camera code (requires Spinnaker SDK already installed)." OFF)
G
gineshidalgo99 已提交
212
# option(WITH_3D_ADAM_MODEL "Add 3-D Adam model (requires OpenGL, Ceres, Eigen, OpenMP, FreeImage, GLEW, and IGL already installed)." OFF)
213

G
gineshidalgo99 已提交
214
# Faster GUI rendering
215 216 217
# Note: It seems to work by default in Windows and Ubuntu, but not in Mac nor Android.
# More info: https://stackoverflow.com/questions/21129683/does-opengl-display-image-faster-than-opencv?answertab=active#tab-top
option(WITH_OPENCV_WITH_OPENGL "Much faster GUI display, but you must also enable OpenGL support in OpenCV by configuring OpenCV using CMake with WITH_OPENGL=ON flag." OFF)
G
gineshidalgo99 已提交
218

219
# Set the acceleration library
G
gineshidalgo99 已提交
220
if (WIN32 OR APPLE)
221 222 223 224 225
  set(WITH_EIGEN NONE CACHE STRING "Select the Eigen mode (NONE if not required or DOWNLOAD to let OpenPose download it).")
  set_property(CACHE WITH_EIGEN PROPERTY STRINGS NONE BUILD)
elseif (UNIX AND NOT APPLE)
  set(WITH_EIGEN NONE CACHE STRING "Select the Eigen mode (NONE if not required, APT_GET if you used `sudo apt-get install libeigen3-dev`, BUILD to let OpenPose download it).")
  set_property(CACHE WITH_EIGEN PROPERTY STRINGS NONE APT_GET BUILD)
G
gineshidalgo99 已提交
226
endif (WIN32 OR APPLE)
227

G
gineshidalgo99 已提交
228 229 230 231 232
# # Suboptions for OpenPose tracking
# if (UNIX AND NOT APPLE)
#   option(WITH_TRACKING "Add OpenPose 3D tracking module (it requires OpenCV with CUDA support)." OFF)
# endif (UNIX AND NOT APPLE)

B
Bikramjot Hanzra 已提交
233
# Download the models
234 235 236
option(DOWNLOAD_BODY_25_MODEL "Download body 25-keypoint (body COCO and 6-keypoint foot) model." ON)
option(DOWNLOAD_BODY_COCO_MODEL "Download body 18-keypoint COCO model." OFF)
option(DOWNLOAD_BODY_MPI_MODEL "Download body 15-keypoint MPI model." OFF)
B
Bikramjot Hanzra 已提交
237
option(DOWNLOAD_FACE_MODEL "Download face model." ON)
238 239 240
option(DOWNLOAD_HAND_MODEL "Download hand model." ON)

# Asio
G
gineshidalgo99 已提交
241
# option(USE_ASIO "Include Asio header-only library." OFF)
B
Bikramjot Hanzra 已提交
242 243 244 245

# More options
option(BUILD_EXAMPLES "Build OpenPose examples." ON)
option(BUILD_DOCS "Build OpenPose documentation." OFF)
R
Raaj 已提交
246
option(BUILD_PYTHON "Build OpenPose python." OFF)
R
Raaj 已提交
247
if (WIN32)
G
gineshidalgo99 已提交
248
  option(BUILD_BIN_FOLDER "Copy all required 3rd-party DLL files into {build_directory}/bin. Disable to save some memory." ON)
R
Raaj 已提交
249
endif ()
B
Bikramjot Hanzra 已提交
250

B
bikz05 已提交
251
# Build as shared library
R
Raaj 已提交
252
option(BUILD_SHARED_LIBS "Build as shared lib." ON)
B
bikz05 已提交
253

254 255 256
# Speed profiler
option(PROFILER_ENABLED "If enabled, OpenPose will be able to print out speed information at runtime." OFF)

G
gineshidalgo99 已提交
257
# Threads - Pthread
258
if (${GPU_MODE} MATCHES "OPENCL" OR (UNIX OR APPLE))
259 260
  unset(CMAKE_THREAD_LIBS_INIT CACHE)
  find_package(Threads)
261 262 263 264
endif (${GPU_MODE} MATCHES "OPENCL" OR (UNIX OR APPLE))

# Caffe OpenCL Boost Issue
if (APPLE)
R
Raaj 已提交
265
  if (${GPU_MODE} MATCHES "OPENCL" OR BUILD_PYTHON)
266 267 268 269 270
    unset(Boost_SYSTEM_LIBRARY CACHE)
    find_package(Boost COMPONENTS system REQUIRED)
  else ()
    set(Boost_SYSTEM_LIBRARY "")
  endif ()
G
gineshidalgo99 已提交
271
endif (APPLE)
B
Bikramjot Hanzra 已提交
272

G
gineshidalgo99 已提交
273
### FIND REQUIRED PACKAGES
B
Bikramjot Hanzra 已提交
274
list(APPEND CMAKE_MODULE_PATH "${CMAKE_CURRENT_SOURCE_DIR}/cmake/Modules")
275

G
gineshidalgo99 已提交
276 277
if (${GPU_MODE} MATCHES "CUDA")
  find_package(CUDA)
278
endif (${GPU_MODE} MATCHES "CUDA")
G
gineshidalgo99 已提交
279

280
# Adding 3D
G
gineshidalgo99 已提交
281 282 283 284
if (WITH_OPENCV_WITH_OPENGL)
  # OpenPose flags
  add_definitions(-DUSE_OPENCV_WITH_OPENGL)
endif (WITH_OPENCV_WITH_OPENGL)
285
if (WITH_3D_RENDERER)
286
  # OpenPose flags
287
  add_definitions(-DUSE_3D_RENDERER)
288
endif (WITH_3D_RENDERER)
289
if (WITH_CERES)
290
  add_definitions(-DUSE_CERES)
291
endif (WITH_CERES)
292 293
if (WITH_FLIR_CAMERA)
  # OpenPose flags
294
  add_definitions(-DUSE_FLIR_CAMERA)
295
endif (WITH_FLIR_CAMERA)
296 297 298 299
if (WITH_3D_ADAM_MODEL)
  # OpenPose flags
  add_definitions(-DUSE_3D_ADAM_MODEL)
endif (WITH_3D_ADAM_MODEL)
300

G
gineshidalgo99 已提交
301 302 303
# Adding tracking
if (WITH_TRACKING)
  # OpenPose flags
304
  add_definitions(-DUSE_TRACKING)
G
gineshidalgo99 已提交
305 306
endif (WITH_TRACKING)

307 308 309 310 311 312 313 314 315
# Calibration
# No Eigen
if (${WITH_EIGEN} MATCHES "NONE")
  if (WITH_CERES)
    message(FATAL_ERROR "Eigen is required (enable `WITH_EIGEN`) if CERES is enabled.")
  endif (WITH_CERES)
# If Eigen used
else (${WITH_EIGEN} MATCHES "NONE")
  # OpenPose flags
316
  add_definitions(-DUSE_EIGEN)
317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332
  # OpenPose download/builds Eigen
  if (${WITH_EIGEN} MATCHES "BUILD")
    # Download it
    set(OP_URL "http://posefs1.perception.cs.cmu.edu/OpenPose/3rdparty/")
    set(FIND_LIB_PREFIX ${CMAKE_SOURCE_DIR}/3rdparty/)
    download_zip("eigen_2018_05_23.zip" ${OP_URL} ${FIND_LIB_PREFIX} 29B9B2FD4679D587BB67467F09EE8365)
    # Set path
    set(EIGEN3_INCLUDE_DIRS "3rdparty/eigen/include/")
  # Alread installed with apt-get
  elseif (${WITH_EIGEN} MATCHES "APT_GET")
    # Eigen
    find_package(PkgConfig)
    pkg_check_modules(EIGEN3 REQUIRED eigen3)
  endif (${WITH_EIGEN} MATCHES "BUILD")
endif (${WITH_EIGEN} MATCHES "NONE")

333
if (UNIX OR APPLE)
G
gineshidalgo99 已提交
334 335 336
  if (${GPU_MODE} MATCHES "CUDA")
    include(cmake/Cuda.cmake)
    find_package(CuDNN)
337
  endif (${GPU_MODE} MATCHES "CUDA")
338 339
  find_package(GFlags)
  find_package(Glog)
B
Bikramjot Hanzra 已提交
340 341 342 343 344
  if (OpenCV_CONFIG_FILE)
    include (${OpenCV_CONFIG_FILE})
  elseif (OpenCV_INCLUDE_DIRS AND OpenCV_LIBS_DIR)
    file(GLOB_RECURSE OpenCV_LIBS "${OpenCV_LIBS_DIR}*.so")
    set(OpenCV_FOUND 1)
345
  else (OpenCV_CONFIG_FILE)
B
Bikramjot Hanzra 已提交
346 347
    find_package(OpenCV)
  endif (OpenCV_CONFIG_FILE)
348

349
  # 3D
350
  if (WITH_3D_RENDERER)
351 352
    # GLUT
    find_package(GLUT REQUIRED)
353
  endif (WITH_3D_RENDERER)
354 355 356 357
  if (WITH_CERES)
    # Eigen + Ceres
    find_package(Ceres REQUIRED COMPONENTS SuiteSparse)
  endif (WITH_CERES)
358
  if (WITH_FLIR_CAMERA)
359 360 361
    # Spinnaker
    find_package(Spinnaker)
    if (NOT SPINNAKER_FOUND)
362 363
      message(FATAL_ERROR "Spinnaker not found. Either turn off the `WITH_FLIR_CAMERA` option or specify the path to
        the Spinnaker includes and libs.")
364
    endif (NOT SPINNAKER_FOUND)
365
  endif (WITH_FLIR_CAMERA)
366
  if (WITH_3D_ADAM_MODEL)
G
gineshidalgo99 已提交
367 368 369
    if (NOT WITH_3D_RENDERER)
      message(FATAL_ERROR "WITH_3D_RENDERER is required if WITH_3D_ADAM_MODEL is enabled.")
    endif (WITH_3D_RENDERER)
370 371
    find_package(PkgConfig)
    pkg_check_modules(EIGEN3 REQUIRED eigen3)
G
gineshidalgo99 已提交
372 373
    # Others: sudo apt-get install libglm-dev
    # http://ceres-solver.org
374
    find_package(Ceres REQUIRED COMPONENTS SuiteSparse)
G
gineshidalgo99 已提交
375
    # sudo apt-get install libglew-dev
376
    find_package(GLEW REQUIRED)
G
gineshidalgo99 已提交
377 378 379 380
    # find_package(GLUT REQUIRED) # TODO: Duplicated of WITH_3D_RENDERER, clean somehow (like Eigen)
    # git clone --recursive https://github.com/libigl/libigl.git
    # No installation, it's header only
    # TODO: It's header only (as Eigen), do BUILD option too to download it
381 382
    find_package(LIBIGL REQUIRED)
    find_package(OpenMP REQUIRED)
G
gineshidalgo99 已提交
383 384
    # Only adam/renderer::Renderer::IdleSaveImage() uses it. Make dependency optional in hand_model
    # FIND_LIBRARY(FREE_IMAGE_LIBRARY NAMES libfreeimage.so)
385 386 387 388
    set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} ${OpenMP_C_FLAGS}")
    set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${OpenMP_CXX_FLAGS}")
    set(CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} ${OpenMP_EXE_LINKER_FLAGS}")
  endif (WITH_3D_ADAM_MODEL)
389

G
gineshidalgo99 已提交
390 391 392 393
  # OpenMP
  if (${GPU_MODE} MATCHES "CPU_ONLY")
    find_package(OpenMP)
    if (OPENMP_FOUND)
G
gineshidalgo99 已提交
394 395 396
      set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} ${OpenMP_C_FLAGS}")
      set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${OpenMP_CXX_FLAGS}")
      set(CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} ${OpenMP_EXE_LINKER_FLAGS}")
397 398
    endif (OPENMP_FOUND)
  endif (${GPU_MODE} MATCHES "CPU_ONLY")
G
gineshidalgo99 已提交
399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414

  if (${GPU_MODE} MATCHES "CUDA")
    # Set CUDA Flags
    set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS} -std=c++11")

    if (NOT CUDA_FOUND)
      message(STATUS "CUDA not found.")
      execute_process(COMMAND cat install_cuda.sh WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/ubuntu)
      message(FATAL_ERROR "Install CUDA using the above commands.")
    endif (NOT CUDA_FOUND)

    if (USE_CUDNN AND NOT CUDNN_FOUND)
      message(STATUS "cuDNN not found.")
      execute_process(COMMAND cat install_cudnn.sh WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/ubuntu)
      message(FATAL_ERROR "Install cuDNN using the above commands. or turn off cuDNN by setting USE_CUDNN to OFF.")
    endif (USE_CUDNN AND NOT CUDNN_FOUND)
415
  endif (${GPU_MODE} MATCHES "CUDA")
416 417

  if (NOT GLOG_FOUND)
G
gineshidalgo99 已提交
418 419
    message(FATAL_ERROR "Glog not found. Install Glog from the command line using the command(s) -\
      sudo apt-get install libgoogle-glog-dev")
420 421 422
  endif (NOT GLOG_FOUND)

  if (NOT GFLAGS_FOUND)
G
gineshidalgo99 已提交
423
    message(FATAL_ERROR "GFlags not found. Install GFlags from the command line using the command(s) --\
424 425 426 427 428 429 430 431
      sudo apt-get install libgflags-dev")
  endif (NOT GFLAGS_FOUND)

  if (NOT OpenCV_FOUND)
    message(FATAL_ERROR "OpenCV not found. Install OpenCV from the command line using the command(s) --\
      sudo apt-get install libopencv-dev")
  endif (NOT OpenCV_FOUND)

432
endif (UNIX OR APPLE)
433 434 435

if (WIN32)

G
gineshidalgo99 已提交
436
  if ("${CMAKE_VERSION}" VERSION_GREATER 3.6.3)
437
    set_property(DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR} PROPERTY VS_STARTUP_PROJECT OpenPoseDemo)
G
gineshidalgo99 已提交
438
  endif ("${CMAKE_VERSION}" VERSION_GREATER 3.6.3)
439 440 441 442 443 444 445 446 447

  set_property(GLOBAL PROPERTY USE_FOLDERS ON)
  set(FIND_LIB_PREFIX ${CMAKE_SOURCE_DIR}/3rdparty/windows)

  # Download Windows 3rd party
  message(STATUS "Downloading windows dependencies...")
  set(OP_WIN_URL "http://posefs1.perception.cs.cmu.edu/OpenPose/3rdparty/windows")
  set(OP_WIN_DIR "${CMAKE_SOURCE_DIR}/3rdparty/windows")

448
  # Download required zip files
449
  download_zip("opencv_310.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} 1e5240a64b814b3c0b822f136be78ad7)
G
gineshidalgo99 已提交
450
  download_zip("caffe3rdparty_2018_06_26.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} 892A39C0CFBAA11CA8648B125852E01F)
451 452
  if (${GPU_MODE} MATCHES "OPENCL")
    download_zip("caffe_opencl_2018_02_13.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} 3ac3e1acf5ee6a4e57920be73053067a)
453 454
  elseif (${GPU_MODE} MATCHES "CPU_ONLY")
    download_zip("caffe_cpu_2018_05_27.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} 87E8401B6DFBAC5B8E909DD20E3B3390)
455
  else (${GPU_MODE} MATCHES "OPENCL")
456
    download_zip("caffe_2018_01_18.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} 4b8e548cc7ea20abea472950dd5301bd)
457
  endif (${GPU_MODE} MATCHES "OPENCL")
458
  if (WITH_3D_RENDERER)
459
    download_zip("freeglut_2018_01_14.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} BB182187285E06880F0EDE3A39530091)
460
  endif (WITH_3D_RENDERER)
461 462 463 464 465 466 467 468
  message(STATUS "Windows dependencies downloaded.")

  find_library(OpenCV_LIBS opencv_world310 HINTS ${FIND_LIB_PREFIX}/opencv/x64/vc14/lib)
  find_library(GFLAGS_LIBRARY_RELEASE gflags HINTS ${FIND_LIB_PREFIX}/caffe3rdparty/lib)
  find_library(GFLAGS_LIBRARY_DEBUG gflagsd HINTS ${FIND_LIB_PREFIX}/caffe3rdparty/lib)
  find_library(GLOG_LIBRARY_RELEASE glog HINTS ${FIND_LIB_PREFIX}/caffe3rdparty/lib)
  find_library(GLOG_LIBRARY_DEBUG glogd HINTS ${FIND_LIB_PREFIX}/caffe3rdparty/lib)
  find_library(OpenCV_LIBS opencv_world310 HINTS ${FIND_LIB_PREFIX}/opencv/x64/vc14/lib)
469

470 471 472
  # If OpenPose builds it
  if (BUILD_CAFFE)
    unset(Caffe_INCLUDE_DIRS CACHE)
473 474
    unset(Caffe_LIB CACHE)
    unset(Caffe_Proto_LIB CACHE)
475 476 477 478
  endif (BUILD_CAFFE)
  # OpenCL
  if (${GPU_MODE} MATCHES "OPENCL")
    set(VCXPROJ_FILE_GPU_MODE "_CL")
479 480
    find_library(Caffe_LIB caffe HINTS ${FIND_LIB_PREFIX}/caffe_opencl/lib)
    find_library(Caffe_Proto_LIB caffeproto HINTS ${FIND_LIB_PREFIX}/caffe_opencl/lib)
481
  # CPU & CUDA
482
  else (${GPU_MODE} MATCHES "OPENCL")
483 484 485 486 487 488 489 490 491 492 493
    # CPU
    if (${GPU_MODE} MATCHES "CPU_ONLY")
      set(VCXPROJ_FILE_GPU_MODE "_CPU")
      find_library(Caffe_LIB caffe HINTS ${FIND_LIB_PREFIX}/caffe_cpu/lib)
      find_library(Caffe_Proto_LIB caffeproto HINTS ${FIND_LIB_PREFIX}/caffe_cpu/lib)
    # CUDA
    else (${GPU_MODE} MATCHES "CPU_ONLY")
      set(VCXPROJ_FILE_GPU_MODE "")
      find_library(Caffe_LIB caffe HINTS ${FIND_LIB_PREFIX}/caffe/lib)
      find_library(Caffe_Proto_LIB caffeproto HINTS ${FIND_LIB_PREFIX}/caffe/lib)
    endif (${GPU_MODE} MATCHES "CPU_ONLY")
494
  endif (${GPU_MODE} MATCHES "OPENCL")
R
Raaj 已提交
495 496 497 498 499 500 501 502 503 504 505 506
  # Boost DepCopy over required DLL F
  if (${GPU_MODE} MATCHES "CPU_ONLY" OR ${GPU_MODE} MATCHES "OPENCL" OR BUILD_PYTHON)
      find_library(BOOST_SYSTEM_LIB_RELEASE libboost_system-vc140-mt-1_61 HINTS ${FIND_LIB_PREFIX}/caffe3rdparty/lib)
      find_library(BOOST_SYSTEM_LIB_DEBUG libboost_system-vc140-mt-gd-1_61 HINTS ${FIND_LIB_PREFIX}/caffe3rdparty/lib)
      find_library(BOOST_FILESYSTEM_LIB_RELEASE libboost_filesystem-vc140-mt-1_61 HINTS ${FIND_LIB_PREFIX}/caffe3rdparty/lib)
      find_library(BOOST_FILESYSTEM_LIB_DEBUG libboost_filesystem-vc140-mt-gd-1_61 HINTS ${FIND_LIB_PREFIX}/caffe3rdparty/lib)
  else ()
      set(BOOST_SYSTEM_LIB_RELEASE "")
      set(BOOST_SYSTEM_LIB_DEBUG "")
      set(BOOST_FILESYSTEM_LIB_RELEASE "")
      set(BOOST_FILESYSTEM_LIB_DEBUG "")
  endif ()
507
  if (WITH_3D_RENDERER)
508 509
    find_library(GLUT_LIBRARY freeglut HINTS ${FIND_LIB_PREFIX}/freeglut/lib)
    message(STATUS "\${GLUT_LIBRARY} = ${GLUT_LIBRARY}")
510 511 512 513
  endif (WITH_3D_RENDERER)
  if (WITH_FLIR_CAMERA)
    find_library(SPINNAKER_LIB spinnaker_v140 HINTS ${FIND_LIB_PREFIX}/spinnaker/lib)
  endif (WITH_FLIR_CAMERA)
514 515
  set(Caffe_LIBS ${Caffe_LIB};${Caffe_Proto_LIB})
  set(OpenCV_INCLUDE_DIRS "3rdparty/windows/opencv/include")
516
  # OpenCL
517 518 519
  if (${GPU_MODE} MATCHES "OPENCL")
    unset(Caffe_INCLUDE_DIRS CACHE)
    set(Caffe_INCLUDE_DIRS "3rdparty/windows/caffe_opencl/include;3rdparty/windows/caffe_opencl/include2" CACHE FILEPATH "Caffe_INCLUDE_DIRS")
520
  # CUDA and CPU
521
  else (${GPU_MODE} MATCHES "OPENCL")
522 523 524 525 526 527 528
    # CPU
    if (${GPU_MODE} MATCHES "CPU_ONLY")
      set(Caffe_INCLUDE_DIRS "3rdparty/windows/caffe_cpu/include;3rdparty/windows/caffe_cpu/include2" CACHE FILEPATH "Caffe_INCLUDE_DIRS")
    # CUDA
    else (${GPU_MODE} MATCHES "CPU_ONLY")
      set(Caffe_INCLUDE_DIRS "3rdparty/windows/caffe/include;3rdparty/windows/caffe/include2" CACHE FILEPATH "Caffe_INCLUDE_DIRS")
    endif (${GPU_MODE} MATCHES "CPU_ONLY")
529
  endif (${GPU_MODE} MATCHES "OPENCL")
530 531
  set(Boost_INCLUDE_DIRS "3rdparty/windows/caffe3rdparty/include/boost-1_61")
  set(WINDOWS_INCLUDE_DIRS "3rdparty/windows/caffe3rdparty/include")
532
  if (WITH_3D_RENDERER)
533
    set(GLUT_INCLUDE_DIRS "3rdparty/windows/freeglut/include")
534 535 536 537
  endif (WITH_3D_RENDERER)
  if (WITH_FLIR_CAMERA)
    set(SPINNAKER_INCLUDE_DIRS "3rdparty/windows/spinnaker/include")
  endif (WITH_FLIR_CAMERA)
B
Bikramjot Hanzra 已提交
538
  set(Caffe_FOUND 1)
R
Raaj 已提交
539 540 541

  # Build DLL Must be on if Build Python is on
  if (BUILD_PYTHON)
G
gineshidalgo99 已提交
542 543
    if (NOT BUILD_BIN_FOLDER)
      message(FATAL_ERROR "BUILD_BIN_FOLDER must be turned on to as well to build python library")
R
Raaj 已提交
544 545 546 547
    endif ()
  endif ()

  # Auto copy DLLs
G
gineshidalgo99 已提交
548 549 550
  if (BUILD_BIN_FOLDER)
    # Locate DLLs
    # Caffe DLLs
R
Raaj 已提交
551 552 553 554 555 556 557
    if (${GPU_MODE} MATCHES "CUDA")
      file(GLOB CAFFE_DLL "${CMAKE_SOURCE_DIR}/3rdparty/windows/caffe/bin/*.dll")
    elseif (${GPU_MODE} MATCHES "OPENCL")
      file(GLOB CAFFE_DLL "${CMAKE_SOURCE_DIR}/3rdparty/windows/caffe_opencl/bin/*.dll")
    elseif (${GPU_MODE} MATCHES "CPU_ONLY")
      file(GLOB CAFFE_DLL "${CMAKE_SOURCE_DIR}/3rdparty/windows/caffe_cpu/bin/*.dll")
    endif ()
G
gineshidalgo99 已提交
558
    # Caffe 3rd-party DLLs
559
    file(GLOB CAFFE_3RD_PARTY_DLL "${CMAKE_SOURCE_DIR}/3rdparty/windows/caffe3rdparty/lib/*.dll")
G
gineshidalgo99 已提交
560 561
    # OpenCV DLLs
    file(GLOB OPENCV_DLL "${CMAKE_SOURCE_DIR}/3rdparty/windows/opencv/x64/vc14/bin/*.dll")
562 563 564 565
    # GLUT DLLs
    file(GLOB GLUT_DLL "${CMAKE_SOURCE_DIR}/3rdparty/windows/freeglut/bin/*.dll")
    # Spinnaker DLLs and other files
    file(GLOB SPINNAKER_DLL "${CMAKE_SOURCE_DIR}/3rdparty/windows/spinnaker/bin/*")
G
gineshidalgo99 已提交
566 567 568 569 570
    # Copy DLLs into same folder
    set(BIN_FOLDER ${CMAKE_BINARY_DIR}/bin)
    file(MAKE_DIRECTORY ${BIN_FOLDER})
    file(COPY ${CAFFE_DLL} DESTINATION ${BIN_FOLDER})
    file(COPY ${OPENCV_DLL} DESTINATION ${BIN_FOLDER})
571 572 573
    file(COPY ${CAFFE_3RD_PARTY_DLL} DESTINATION ${BIN_FOLDER})
    file(COPY ${GLUT_DLL} DESTINATION ${BIN_FOLDER})
    file(COPY ${SPINNAKER_DLL} DESTINATION ${BIN_FOLDER})
G
gineshidalgo99 已提交
574
  endif (BUILD_BIN_FOLDER)
R
Raaj 已提交
575

576
endif (WIN32)
B
Bikramjot Hanzra 已提交
577 578


G
gineshidalgo99 已提交
579
### CAFFE
580
if (UNIX OR APPLE)
B
Bikramjot Hanzra 已提交
581

582
  if (${DL_FRAMEWORK} MATCHES "CAFFE")
B
Bikramjot Hanzra 已提交
583

584 585 586 587 588 589
    # Check if the user specified caffe paths
    if (Caffe_INCLUDE_DIRS AND Caffe_LIBS AND NOT BUILD_CAFFE)
      message(STATUS "\${Caffe_INCLUDE_DIRS} set by the user to " ${Caffe_INCLUDE_DIRS})
      message(STATUS "\${Caffe_LIBS} set by the user to " ${Caffe_LIBS})
      set(Caffe_FOUND 1)
    endif (Caffe_INCLUDE_DIRS AND Caffe_LIBS AND NOT BUILD_CAFFE)
B
Bikramjot Hanzra 已提交
590

591 592
    # Else build from scratch
    if (BUILD_CAFFE)
B
Bikramjot Hanzra 已提交
593

594 595
      # Download Caffe
      message(STATUS "Caffe will be downloaded from source now. NOTE: This process might take several minutes depending
G
gineshidalgo99 已提交
596 597 598 599 600 601 602
        on your internet connection.")

      # Check if pulled
      file(GLOB CAFFE_DIR_VALID ${CMAKE_SOURCE_DIR}/3rdparty/caffe/*)
      list(LENGTH CAFFE_DIR_VALID CAFFE_DIR_VALID_LENGTH)
      if (CAFFE_DIR_VALID_LENGTH EQUAL 0)
        execute_process(COMMAND git submodule update --init --recursive)
603
      else (CAFFE_DIR_VALID_LENGTH EQUAL 0)
G
gineshidalgo99 已提交
604
        message(STATUS "Caffe has already been downloaded.")
605
      endif (CAFFE_DIR_VALID_LENGTH EQUAL 0)
G
gineshidalgo99 已提交
606 607 608 609 610 611 612

      # Build Process
      set(CAFFE_CPU_ONLY OFF)
      if (${GPU_MODE} MATCHES "CUDA")
        execute_process(COMMAND git checkout master WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
      elseif (${GPU_MODE} MATCHES "CPU_ONLY")
        if (USE_MKL)
G
gineshidalgo99 已提交
613 614
          #execute_process(COMMAND git checkout intel WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
          execute_process(COMMAND git checkout b6712ce WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
B
Bikramjot Hanzra 已提交
615 616
          execute_process(COMMAND sh prepare_mkl.sh WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/3rdparty/caffe/external/mkl
              OUTPUT_VARIABLE rv)
G
gineshidalgo99 已提交
617 618 619
          set( MLIST ${rv} )
          separate_arguments(MLIST)
          list(GET MLIST 0 MKL_PATH)
620
          message(STATUS ${MKL_PATH})
G
gineshidalgo99 已提交
621 622 623 624 625
          file(GLOB MKL_SO
            "${MKL_PATH}lib/*"
            )
          file(COPY ${MKL_SO} DESTINATION ${CMAKE_BINARY_DIR}/caffe)

G
gineshidalgo99 已提交
626 627 628 629 630 631 632 633 634 635 636 637
          # New MLSL Lib
          #execute_process(COMMAND sh prepare_mlsl.sh WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/3rdparty/caffe/external/mlsl
          #    OUTPUT_VARIABLE rv)
          #set( MLIST ${rv} )
          #separate_arguments(MLIST)
          #list(GET MLIST 0 MLSL_PATH)
          #message(STATUS ${MLSL_PATH})
          #file(GLOB MLSL_SO
          #  "${MLSL_PATH}/intel64/lib/*"
          #  )
          #file(COPY ${MLSL_SO} DESTINATION ${CMAKE_BINARY_DIR}/caffe)

G
gineshidalgo99 已提交
638
          set(MKL_LIBS
G
gineshidalgo99 已提交
639
            #"${CMAKE_BINARY_DIR}/caffe/libmlsl.so"
G
gineshidalgo99 已提交
640 641 642 643
            "${CMAKE_BINARY_DIR}/caffe/libiomp5.so"
            "${CMAKE_BINARY_DIR}/caffe/libmklml_intel.so"
            "${CMAKE_BINARY_DIR}/caffe/lib/libmkldnn.so"
            )
644
        else (USE_MKL)
G
gineshidalgo99 已提交
645
          execute_process(COMMAND git checkout master WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
646
        endif (USE_MKL)
G
gineshidalgo99 已提交
647 648
        set(CAFFE_CPU_ONLY ON)
        set(USE_CUDNN OFF)
649 650 651
      elseif (${GPU_MODE} MATCHES "OPENCL")
        execute_process(COMMAND git checkout opencl WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
        set(USE_CUDNN OFF)
652
      endif (${GPU_MODE} MATCHES "CUDA")
G
gineshidalgo99 已提交
653

654 655
      # Build Caffe
      message(STATUS "Caffe will be built from source now.")
B
Bikramjot Hanzra 已提交
656
      find_package(Caffe)
657 658 659
      include(ExternalProject)
      set(CAFFE_PREFIX caffe)
      set(CAFFE_URL ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
B
Bikramjot Hanzra 已提交
660

G
gineshidalgo99 已提交
661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679
      # One for Intel Branch and one for Master
      if (USE_MKL)
        ExternalProject_Add(openpose_caffe
          SOURCE_DIR ${CAFFE_URL}
          PREFIX ${CAFFE_PREFIX}
          CMAKE_ARGS -DCMAKE_INSTALL_PREFIX:PATH=<INSTALL_DIR>
          -DMKLDNN_INSTALL_DIR:PATH=<INSTALL_DIR>
          -DUSE_CUDNN=${USE_CUDNN}
          -DBUILD_python=OFF
          -DCPU_ONLY=${CAFFE_CPU_ONLY}
          -DUSE_MKL2017_AS_DEFAULT_ENGINE=${CAFFE_CPU_ONLY}
          -DCMAKE_BUILD_TYPE=Release
          -DBUILD_docs=OFF
          -DBUILD_python=OFF
          -DBUILD_python_layer=OFF
          -DUSE_LEVELDB=OFF
          -DUSE_LMDB=OFF
          -DUSE_OPENCV=OFF)
          # -DOpenCV_DIR=${OpenCV_DIR})
680
      else (USE_MKL)
G
gineshidalgo99 已提交
681
        ExternalProject_Add(openpose_caffe
682 683
          SOURCE_DIR ${CAFFE_URL}
          PREFIX ${CAFFE_PREFIX}
G
gineshidalgo99 已提交
684
          CMAKE_ARGS -DCMAKE_INSTALL_PREFIX:PATH=<INSTALL_DIR>
G
gineshidalgo99 已提交
685 686 687 688 689 690 691 692 693 694
          -DUSE_CUDNN=${USE_CUDNN}
          -DCPU_ONLY=${CAFFE_CPU_ONLY}
          -DCMAKE_BUILD_TYPE=Release
          -DBUILD_docs=OFF
          -DBUILD_python=OFF
          -DBUILD_python_layer=OFF
          -DUSE_LEVELDB=OFF
          -DUSE_LMDB=OFF
          -DUSE_OPENCV=OFF)
          # -DOpenCV_DIR=${OpenCV_DIR})
695
      endif (USE_MKL)
B
Bikramjot Hanzra 已提交
696

697 698
      ExternalProject_Get_Property(openpose_caffe install_dir)

B
Bikramjot Hanzra 已提交
699 700 701
      if (NOT Caffe_FOUND)
        add_custom_command(TARGET openpose_caffe
            POST_BUILD
702
            COMMAND ${CMAKE_COMMAND} ${CMAKE_SOURCE_DIR}
B
Bikramjot Hanzra 已提交
703 704 705 706
            COMMAND $(MAKE)
            WORKING_DIRECTORY ${CMAKE_BINARY_DIR}
            COMMENT "Rerunning cmake after building Caffe submodule")
      endif (NOT Caffe_FOUND)
707 708 709 710

    endif (BUILD_CAFFE)

    if (NOT Caffe_FOUND AND NOT BUILD_CAFFE)
G
gineshidalgo99 已提交
711
      message(FATAL_ERROR "Caffe not found. Either turn on the BUILD_CAFFE option or specify the path of Caffe includes
712
        and libs using -DCaffe_INCLUDE_DIRS and -DCaffe_LIBS.")
713 714 715 716
    endif (NOT Caffe_FOUND AND NOT BUILD_CAFFE)

  endif (${DL_FRAMEWORK} MATCHES "CAFFE")

717
endif (UNIX OR APPLE)
B
Bikramjot Hanzra 已提交
718 719 720 721

### PROJECT INCLUDES
# Specify the include directories
include_directories(
722
  include
B
Bikramjot Hanzra 已提交
723 724 725
  ${GFLAGS_INCLUDE_DIR}
  ${GLOG_INCLUDE_DIR}
  ${OpenCV_INCLUDE_DIRS})
B
Bikramjot Hanzra 已提交
726

727 728 729 730 731 732 733 734
if (USE_ASIO)
  include_directories(${CMAKE_SOURCE_DIR}/3rdparty/asio/include/)
  # OpenPose flags
  add_definitions(-DUSE_ASIO)
  # Tell Asio it is not using Boost
  add_definitions(-DASIO_STANDALONE)
endif (USE_ASIO)

735 736 737 738 739 740
# Calibration
if (NOT ${WITH_EIGEN} MATCHES "NONE")
  include_directories(
      ${EIGEN3_INCLUDE_DIRS})
endif (NOT ${WITH_EIGEN} MATCHES "NONE")

741 742 743 744 745
if (APPLE)
  include_directories(
      "/usr/local/opt/openblas/include")
endif (APPLE)

746
if (USE_MKL)
747 748 749 750
 include_directories(
     "${MKL_PATH}/include/")
endif (USE_MKL)

B
Bikramjot Hanzra 已提交
751 752 753 754 755
if (Caffe_FOUND)
  include_directories(
      ${Caffe_INCLUDE_DIRS})
endif (Caffe_FOUND)

G
gineshidalgo99 已提交
756 757
if (${GPU_MODE} MATCHES "CUDA")
  include_directories(
B
Bikramjot Hanzra 已提交
758
      ${CUDA_INCLUDE_DIRS})
759
elseif (${GPU_MODE} MATCHES "OPENCL")
760 761
  include_directories(
    ${OpenCL_INCLUDE_DIRS})
762
endif (${GPU_MODE} MATCHES "CUDA")
763
# 3D
764 765 766
if (WITH_3D_RENDERER)
  include_directories(${GLUT_INCLUDE_DIRS})
endif (WITH_3D_RENDERER)
767 768 769
if (WITH_CERES)
  include_directories(${CERES_INCLUDE_DIRS})
endif (WITH_CERES)
770 771 772
if (WITH_FLIR_CAMERA)
  include_directories(${SPINNAKER_INCLUDE_DIRS})
endif (WITH_FLIR_CAMERA)
773 774 775 776 777 778 779 780
if (WITH_3D_ADAM_MODEL)
                                    include_directories(include/adam) # TEMPORARY - TO BE REMOVED IN THE FUTURE
  include_directories(${CERES_INCLUDE_DIRS})
  include_directories(${EIGEN3_INCLUDE_DIRS})
  include_directories(${IGL_INCLUDE_DIRS})
  include_directories(${LIBIGL_INCLUDE_DIRS})
  include_directories(${OPENGL_INCLUDE_DIRS} ${GLUT_INCLUDE_DIRS} ${GLEW_INCLUDE_DIRS})
endif (WITH_3D_ADAM_MODEL)
781
# Windows includes
782 783 784 785 786
if (WIN32)
  include_directories(
    ${Boost_INCLUDE_DIRS}
    ${WINDOWS_INCLUDE_DIRS})
endif (WIN32)
B
Bikramjot Hanzra 已提交
787 788


789 790
### COLLECT ALL 3RD-PARTY LIBRARIES TO BE LINKED AGAINST
set(OpenPose_3rdparty_libraries ${OpenCV_LIBS} ${GLOG_LIBRARY})
791
if (UNIX OR APPLE)
792 793 794 795 796
  set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries} ${GLOG_LIBRARY})
elseif (WIN32)
  set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries}
      debug ${GFLAGS_LIBRARY_DEBUG} optimized ${GFLAGS_LIBRARY_RELEASE}
      debug ${GLOG_LIBRARY_DEBUG} optimized ${GLOG_LIBRARY_RELEASE})
797
endif (UNIX OR APPLE)
798 799 800 801 802 803 804 805 806 807 808
# Deep net Framework
if (${DL_FRAMEWORK} MATCHES "CAFFE")
  set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries} ${Caffe_LIBS} ${GFLAGS_LIBRARY})
endif (${DL_FRAMEWORK} MATCHES "CAFFE")
# CPU vs. GPU
if (USE_MKL)
  set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries} ${MKL_LIBS})
endif (USE_MKL)
if (${GPU_MODE} MATCHES "OPENCL")
  set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries} ${CMAKE_THREAD_LIBS_INIT} ${OpenCL_LIBRARIES})
endif (${GPU_MODE} MATCHES "OPENCL")
R
Raaj 已提交
809 810 811 812 813 814 815 816 817
# Boost
if (WIN32)
    if (${GPU_MODE} MATCHES "CPU_ONLY" OR ${GPU_MODE} MATCHES "OPENCL" OR BUILD_PYTHON)
    set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries}
        debug ${BOOST_SYSTEM_LIB_RELEASE} optimized ${BOOST_SYSTEM_LIB_RELEASE})
        set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries}
        debug ${BOOST_FILESYSTEM_LIB_RELEASE} optimized ${BOOST_FILESYSTEM_LIB_RELEASE})
    endif ()
endif (WIN32)
818
# 3-D
819 820 821 822
if (WITH_3D_ADAM_MODEL)
  set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries}
      ${OPENGL_LIBRARIES} ${GLUT_LIBRARY} ${GLEW_LIBRARY} ${FREE_IMAGE_LIBRARY})
endif (WITH_3D_ADAM_MODEL)
823 824 825 826 827 828 829 830 831
if (WITH_3D_RENDERER)
  set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries} ${GLUT_LIBRARY})
endif (WITH_3D_RENDERER)
if (WITH_CERES)
  set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries} ${CERES_LIBRARIES})
endif (WITH_CERES)
if (WITH_FLIR_CAMERA)
  set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries} ${SPINNAKER_LIB})
endif (WITH_FLIR_CAMERA)
G
gineshidalgo99 已提交
832
# Pthread
833
if (UNIX OR APPLE)
G
gineshidalgo99 已提交
834
  set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries} pthread)
835
endif (UNIX OR APPLE)
836 837 838 839

set(examples_3rdparty_libraries ${OpenPose_3rdparty_libraries} ${GFLAGS_LIBRARY})


B
Bikramjot Hanzra 已提交
840 841 842 843 844 845 846
### ADD SUBDIRECTORIES
if (Caffe_FOUND)
  add_subdirectory(src)
  if (BUILD_EXAMPLES)
    add_subdirectory(examples)
  endif (BUILD_EXAMPLES)
endif (Caffe_FOUND)
B
Bikramjot Hanzra 已提交
847 848


G
gineshidalgo99 已提交
849
### DOWNLOAD MODELS
B
Bikramjot Hanzra 已提交
850 851 852 853 854
# Download the models if flag is set
message(STATUS "Download the models.")

# URL to the models
set(OPENPOSE_URL "http://posefs1.perception.cs.cmu.edu/OpenPose/models/")
G
gineshidalgo99 已提交
855

856 857 858
download_model("BODY_25" ${DOWNLOAD_BODY_25_MODEL} pose/body_25/pose_iter_584000.caffemodel
  78287B57CF85FA89C03F1393D368E5B7) # Body (BODY_25)
download_model("body (COCO)" ${DOWNLOAD_BODY_COCO_MODEL} pose/coco/pose_iter_440000.caffemodel
G
gineshidalgo99 已提交
859
  5156d31f670511fce9b4e28b403f2939) # Body (COCO)
860
download_model("body (MPI)" ${DOWNLOAD_BODY_MPI_MODEL} pose/mpi/pose_iter_160000.caffemodel
G
gineshidalgo99 已提交
861
  2ca0990c7562bd7ae03f3f54afa96e00) # Body (MPI)
862
download_model("face" ${DOWNLOAD_FACE_MODEL} face/pose_iter_116000.caffemodel
G
gineshidalgo99 已提交
863
  e747180d728fa4e4418c465828384333) # Face
864
download_model("hand" ${DOWNLOAD_HAND_MODEL} hand/pose_iter_102000.caffemodel
G
gineshidalgo99 已提交
865
  a82cfc3fea7c62f159e11bd3674c1531) # Hand
B
Bikramjot Hanzra 已提交
866 867

message(STATUS "Models Downloaded.")
B
bikz05 已提交
868

R
Raaj 已提交
869 870 871 872 873 874
### PYTHON
if (Caffe_FOUND)
  if(BUILD_PYTHON)
    add_subdirectory(python)
  endif ()
endif()
875

G
gineshidalgo99 已提交
876
### GENERATE DOCUMENTATION
877
if (UNIX OR APPLE)
878 879 880 881 882 883 884 885 886

  if (BUILD_DOCS)
    find_package(Doxygen)
    if (DOXYGEN_FOUND)
      # Set input and output files
      set(DOXYGEN_FILE ${CMAKE_SOURCE_DIR}/doc/doc_autogeneration.doxygen)

      # Custom target to build the documentation
      add_custom_target(doc_doxygen ALL
G
gineshidalgo99 已提交
887 888 889 890
        COMMAND ${DOXYGEN_EXECUTABLE} ${DOXYGEN_FILE}
        WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/doc
        COMMENT "Generating API documentation with Doxygen"
        VERBATIM)
891
    else (DOXYGEN_FOUND)
892
      message(FATAL_ERROR "Doxygen needs to be installed to generate the doxygen documentation.")
893 894 895
    endif (DOXYGEN_FOUND)
  endif (BUILD_DOCS)

896
endif (UNIX OR APPLE)
897

B
bikz05 已提交
898

G
gineshidalgo99 已提交
899
### INSTALL
900
if (UNIX OR APPLE)
B
Bikramjot Hanzra 已提交
901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926
  if (Caffe_FOUND)
    # Install the headers
    install(DIRECTORY ${CMAKE_SOURCE_DIR}/include/openpose DESTINATION include)
    install(EXPORT OpenPose DESTINATION lib/OpenPose)
    if (BUILD_CAFFE)
      install(DIRECTORY ${CMAKE_BINARY_DIR}/caffe/include/caffe DESTINATION include)
      install(DIRECTORY ${CMAKE_BINARY_DIR}/caffe/lib/ DESTINATION lib)
    endif (BUILD_CAFFE)

    # Compute installation prefix relative to this file
    configure_file(
      ${CMAKE_SOURCE_DIR}/cmake/OpenPoseConfig.cmake.in
      ${CMAKE_BINARY_DIR}/cmake/OpenPoseConfig.cmake @ONLY)

    install(FILES ${CMAKE_BINARY_DIR}/cmake/OpenPoseConfig.cmake
      DESTINATION lib/OpenPose)

    # Uninstall target
    configure_file(
      "${CMAKE_SOURCE_DIR}/cmake/cmake_uninstall.cmake.in"
      "${CMAKE_BINARY_DIR}/cmake_uninstall.cmake"
      IMMEDIATE @ONLY)

    add_custom_target(uninstall
        COMMAND ${CMAKE_COMMAND} -P ${CMAKE_CURRENT_BINARY_DIR}/cmake_uninstall.cmake)
  endif (Caffe_FOUND)
927
endif (UNIX OR APPLE)