CMakeLists.txt 34.9 KB
Newer Older
B
bikz05 已提交
1 2
### VERSION INFO
set(OpenPose_VERSION_MAJOR 1)
G
gineshidalgo99 已提交
3
set(OpenPose_VERSION_MINOR 4)
G
gineshidalgo99 已提交
4
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


29 30 31 32 33 34
### BUILD_TYPE
# Default: Release
set(CMAKE_BUILD_TYPE "Release" CACHE STRING "Choose the type of build." FORCE)
set_property(CACHE CMAKE_BUILD_TYPE PROPERTY STRINGS "Debug" "Release" "MinSizeRel" "RelWithDebInfo")


G
gineshidalgo99 已提交
35
### FLAGS
36
if (WIN32)
G
gineshidalgo99 已提交
37
  # TODO -- Makeshift solution -- This prevents rerunning build again
38 39 40
  # https://gitlab.kitware.com/cmake/cmake/issues/16783
  set(CMAKE_SUPPRESS_REGENERATION ON)

A
arkceajin 已提交
41 42
  string (REPLACE "/D_WINDOWS" "" CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}")
  string (REPLACE "/DWIN32" "" CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}")
43 44 45 46

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

G
gineshidalgo99 已提交
47
  set(CMAKE_CONFIGURATION_TYPES Release Debug CACHE TYPE INTERNAL FORCE)
48

49 50 51 52 53 54 55 56 57
  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")
58

59 60 61
  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}")
62
elseif (UNIX)
G
gineshidalgo99 已提交
63
  # Turn on C++11
64 65
  add_definitions(-std=c++11)
  set(CMAKE_CXX_FLAGS_RELEASE "-O3")
66 67 68 69
elseif (APPLE)
  # Turn on C++11
  set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11")
  set(CMAKE_CXX_FLAGS_RELEASE "-O3")
70
endif (WIN32)
B
Bikramjot Hanzra 已提交
71 72

# C++ additional flags
G
gineshidalgo99 已提交
73
if (CMAKE_COMPILER_IS_GNUCXX)
74 75 76 77 78 79 80 81

  # 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")
82
  endif (${INSTRUCTION_SET} MATCHES "SSE4")
83 84 85 86

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

G
gineshidalgo99 已提交
89
  message(STATUS "GCC detected, adding compile flags")
90
  set(OP_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${SIMD_FLAGS} -fopenmp -Wpedantic -Wall -Wextra -Wfatal-errors")
G
gineshidalgo99 已提交
91
endif (CMAKE_COMPILER_IS_GNUCXX)
B
Bikramjot Hanzra 已提交
92 93


G
gineshidalgo99 已提交
94
### PROJECT OPTIONS
B
Bikramjot Hanzra 已提交
95 96 97 98 99 100 101 102 103
# 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 已提交
104
    "DL_FRAMEWORK" ON)
105 106
  # OpenPose flags
  add_definitions(-DUSE_CAFFE)
B
Bikramjot Hanzra 已提交
107 108
endif (${DL_FRAMEWORK} MATCHES "CAFFE")

109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127
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 已提交
128
# Set the acceleration library
G
gineshidalgo99 已提交
129 130 131 132 133
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)
134 135 136 137
# 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 已提交
138
#   set_property(CACHE GPU_MODE PROPERTY STRINGS CUDA CPU_ONLY)
139 140 141 142
# elseif (OpenCL_FOUND)
#   set_property(CACHE GPU_MODE PROPERTY STRINGS OPENCL CPU_ONLY)
# else ()
#   set_property(CACHE GPU_MODE PROPERTY STRINGS CPU_ONLY)
143
# endif (CUDA_FOUND AND OpenCL_FOUND)
R
Raaj 已提交
144
set_property(CACHE GPU_MODE PROPERTY STRINGS CUDA OPENCL CPU_ONLY)
G
gineshidalgo99 已提交
145

146
# Look for CUDA
G
gineshidalgo99 已提交
147
set(CUDA_FOUND FALSE)
148 149 150 151 152
if (${GPU_MODE} MATCHES "CUDA")
  find_package(CUDA)
endif (${GPU_MODE} MATCHES "CUDA")
# Look for OpenCL
set(OpenCL_FOUND FALSE)
G
gineshidalgo99 已提交
153
set(CUDA_VERSION_MAJOR 0)
154 155 156 157
if (${GPU_MODE} MATCHES "OPENCL")
  find_package(OpenCL)
endif (${GPU_MODE} MATCHES "OPENCL")

158 159 160 161 162 163 164
# 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)
165
endif (CUDA_FOUND AND OpenCL_FOUND AND ${CUDA_VERSION_MAJOR} LESS 9)
166
# Handle desired GPU mode option
167 168 169
if (${GPU_MODE} MATCHES "CUDA")
  # OpenPose flags
  add_definitions(-DUSE_CUDA)
G
gineshidalgo99 已提交
170
  message(STATUS "Building with CUDA.")
171
elseif (${GPU_MODE} MATCHES "CPU_ONLY")
G
gineshidalgo99 已提交
172 173
  add_definitions(-DUSE_CPU_ONLY)
  message(STATUS "Building CPU Only.")
174 175
  # OpenPose flag for Caffe
  add_definitions(-DCPU_ONLY)
176 177 178 179 180
elseif (${GPU_MODE} MATCHES "OPENCL")
  # OpenPose flag for Caffe
  add_definitions(-DUSE_OPENCL)
  add_definitions(-DUSE_GREENTEA)
  message(STATUS "Building with OpenCL.")
181
endif (${GPU_MODE} MATCHES "CUDA")
G
gineshidalgo99 已提交
182 183 184 185 186

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

190 191 192 193 194 195
if (${USE_MKL})
  # OpenPose flags
  add_definitions(-DUSE_MKL)
  message(STATUS "Building with MKL support.")
endif (${USE_MKL})

196 197 198 199 200
# Set/disable profiler
if (PROFILER_ENABLED)
  add_definitions(-DPROFILER_ENABLED)
endif (PROFILER_ENABLED)

B
Bikramjot Hanzra 已提交
201 202 203 204 205 206 207 208 209 210
# 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")
211

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

G
gineshidalgo99 已提交
220
# Faster GUI rendering
221 222 223
# 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 已提交
224

225
# Set the acceleration library
G
gineshidalgo99 已提交
226
if (WIN32 OR APPLE)
227 228 229 230 231
  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 已提交
232
endif (WIN32 OR APPLE)
233

G
gineshidalgo99 已提交
234 235 236 237 238
# # 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 已提交
239
# Download the models
240 241 242
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 已提交
243
option(DOWNLOAD_FACE_MODEL "Download face model." ON)
244 245 246
option(DOWNLOAD_HAND_MODEL "Download hand model." ON)

# Asio
G
gineshidalgo99 已提交
247
# option(USE_ASIO "Include Asio header-only library." OFF)
B
Bikramjot Hanzra 已提交
248 249 250 251

# More options
option(BUILD_EXAMPLES "Build OpenPose examples." ON)
option(BUILD_DOCS "Build OpenPose documentation." OFF)
R
Raaj 已提交
252
option(BUILD_PYTHON "Build OpenPose python." OFF)
R
Raaj 已提交
253
if (WIN32)
G
gineshidalgo99 已提交
254
  option(BUILD_BIN_FOLDER "Copy all required 3rd-party DLL files into {build_directory}/bin. Disable to save some memory." ON)
G
gineshidalgo99 已提交
255 256 257 258
endif (WIN32)

# Unity
# option(BUILD_UNITY_SUPPORT "Build OpenPose as a Unity plugin." OFF)
B
Bikramjot Hanzra 已提交
259

B
bikz05 已提交
260
# Build as shared library
R
Raaj 已提交
261
option(BUILD_SHARED_LIBS "Build as shared lib." ON)
B
bikz05 已提交
262

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

G
gineshidalgo99 已提交
266
# Threads - Pthread
267
if (${GPU_MODE} MATCHES "OPENCL" OR (UNIX OR APPLE))
268 269
  unset(CMAKE_THREAD_LIBS_INIT CACHE)
  find_package(Threads)
270 271 272 273
endif (${GPU_MODE} MATCHES "OPENCL" OR (UNIX OR APPLE))

# Caffe OpenCL Boost Issue
if (APPLE)
R
Raaj 已提交
274
  if (${GPU_MODE} MATCHES "OPENCL" OR BUILD_PYTHON)
275 276 277 278 279
    unset(Boost_SYSTEM_LIBRARY CACHE)
    find_package(Boost COMPONENTS system REQUIRED)
  else ()
    set(Boost_SYSTEM_LIBRARY "")
  endif ()
G
gineshidalgo99 已提交
280
endif (APPLE)
B
Bikramjot Hanzra 已提交
281

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

G
gineshidalgo99 已提交
285 286
if (${GPU_MODE} MATCHES "CUDA")
  find_package(CUDA)
287
endif (${GPU_MODE} MATCHES "CUDA")
G
gineshidalgo99 已提交
288

289
# Adding 3D
G
gineshidalgo99 已提交
290 291 292 293
if (WITH_OPENCV_WITH_OPENGL)
  # OpenPose flags
  add_definitions(-DUSE_OPENCV_WITH_OPENGL)
endif (WITH_OPENCV_WITH_OPENGL)
294
if (WITH_3D_RENDERER)
295
  # OpenPose flags
296
  add_definitions(-DUSE_3D_RENDERER)
297
endif (WITH_3D_RENDERER)
298
if (WITH_CERES)
299
  add_definitions(-DUSE_CERES)
300
endif (WITH_CERES)
301 302
if (WITH_FLIR_CAMERA)
  # OpenPose flags
303
  add_definitions(-DUSE_FLIR_CAMERA)
304
endif (WITH_FLIR_CAMERA)
305 306 307 308
if (WITH_3D_ADAM_MODEL)
  # OpenPose flags
  add_definitions(-DUSE_3D_ADAM_MODEL)
endif (WITH_3D_ADAM_MODEL)
309

G
gineshidalgo99 已提交
310 311 312
# Adding tracking
if (WITH_TRACKING)
  # OpenPose flags
313
  add_definitions(-DUSE_TRACKING)
G
gineshidalgo99 已提交
314 315
endif (WITH_TRACKING)

G
gineshidalgo99 已提交
316 317 318 319 320 321
# Unity
if (BUILD_UNITY_SUPPORT)
  # OpenPose flags
  add_definitions(-DUSE_UNITY_SUPPORT)
endif (BUILD_UNITY_SUPPORT)

322 323 324 325 326 327 328 329 330
# 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
331
  add_definitions(-DUSE_EIGEN)
332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347
  # 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")

348
if (UNIX OR APPLE)
G
gineshidalgo99 已提交
349 350 351
  if (${GPU_MODE} MATCHES "CUDA")
    include(cmake/Cuda.cmake)
    find_package(CuDNN)
352
  endif (${GPU_MODE} MATCHES "CUDA")
353 354
  find_package(GFlags)
  find_package(Glog)
B
Bikramjot Hanzra 已提交
355 356 357 358 359
  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)
360
  else (OpenCV_CONFIG_FILE)
B
Bikramjot Hanzra 已提交
361 362
    find_package(OpenCV)
  endif (OpenCV_CONFIG_FILE)
363

364
  # 3D
365
  if (WITH_3D_RENDERER)
366 367
    # GLUT
    find_package(GLUT REQUIRED)
368
  endif (WITH_3D_RENDERER)
369 370 371 372
  if (WITH_CERES)
    # Eigen + Ceres
    find_package(Ceres REQUIRED COMPONENTS SuiteSparse)
  endif (WITH_CERES)
373
  if (WITH_FLIR_CAMERA)
374 375 376
    # Spinnaker
    find_package(Spinnaker)
    if (NOT SPINNAKER_FOUND)
377 378
      message(FATAL_ERROR "Spinnaker not found. Either turn off the `WITH_FLIR_CAMERA` option or specify the path to
        the Spinnaker includes and libs.")
379
    endif (NOT SPINNAKER_FOUND)
380
  endif (WITH_FLIR_CAMERA)
381
  if (WITH_3D_ADAM_MODEL)
G
gineshidalgo99 已提交
382 383 384
    if (NOT WITH_3D_RENDERER)
      message(FATAL_ERROR "WITH_3D_RENDERER is required if WITH_3D_ADAM_MODEL is enabled.")
    endif (WITH_3D_RENDERER)
385 386
    find_package(PkgConfig)
    pkg_check_modules(EIGEN3 REQUIRED eigen3)
G
gineshidalgo99 已提交
387 388
    # Others: sudo apt-get install libglm-dev
    # http://ceres-solver.org
389
    find_package(Ceres REQUIRED COMPONENTS SuiteSparse)
G
gineshidalgo99 已提交
390
    # sudo apt-get install libglew-dev
391
    find_package(GLEW REQUIRED)
G
gineshidalgo99 已提交
392 393 394 395
    # 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
396 397
    find_package(LIBIGL REQUIRED)
    find_package(OpenMP REQUIRED)
G
gineshidalgo99 已提交
398 399
    # Only adam/renderer::Renderer::IdleSaveImage() uses it. Make dependency optional in hand_model
    # FIND_LIBRARY(FREE_IMAGE_LIBRARY NAMES libfreeimage.so)
400 401 402 403
    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)
404

G
gineshidalgo99 已提交
405 406 407 408
  # OpenMP
  if (${GPU_MODE} MATCHES "CPU_ONLY")
    find_package(OpenMP)
    if (OPENMP_FOUND)
G
gineshidalgo99 已提交
409 410 411
      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}")
412 413
    endif (OPENMP_FOUND)
  endif (${GPU_MODE} MATCHES "CPU_ONLY")
G
gineshidalgo99 已提交
414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429

  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)
430
  endif (${GPU_MODE} MATCHES "CUDA")
431 432

  if (NOT GLOG_FOUND)
G
gineshidalgo99 已提交
433 434
    message(FATAL_ERROR "Glog not found. Install Glog from the command line using the command(s) -\
      sudo apt-get install libgoogle-glog-dev")
435 436 437
  endif (NOT GLOG_FOUND)

  if (NOT GFLAGS_FOUND)
G
gineshidalgo99 已提交
438
    message(FATAL_ERROR "GFlags not found. Install GFlags from the command line using the command(s) --\
439 440 441 442 443 444 445 446
      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)

447
endif (UNIX OR APPLE)
448 449 450

if (WIN32)

G
gineshidalgo99 已提交
451
  if ("${CMAKE_VERSION}" VERSION_GREATER 3.6.3)
452
    set_property(DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR} PROPERTY VS_STARTUP_PROJECT OpenPoseDemo)
G
gineshidalgo99 已提交
453
  endif ("${CMAKE_VERSION}" VERSION_GREATER 3.6.3)
454 455 456 457 458 459 460 461 462

  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")

463
  # Download required zip files
464
  download_zip("opencv_310.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} 1e5240a64b814b3c0b822f136be78ad7)
G
gineshidalgo99 已提交
465
  download_zip("caffe3rdparty_2018_06_26.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} 892A39C0CFBAA11CA8648B125852E01F)
466 467
  if (${GPU_MODE} MATCHES "OPENCL")
    download_zip("caffe_opencl_2018_02_13.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} 3ac3e1acf5ee6a4e57920be73053067a)
468 469
  elseif (${GPU_MODE} MATCHES "CPU_ONLY")
    download_zip("caffe_cpu_2018_05_27.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} 87E8401B6DFBAC5B8E909DD20E3B3390)
470
  else (${GPU_MODE} MATCHES "OPENCL")
471
    download_zip("caffe_2018_01_18.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} 4b8e548cc7ea20abea472950dd5301bd)
472
  endif (${GPU_MODE} MATCHES "OPENCL")
473
  if (WITH_3D_RENDERER)
474
    download_zip("freeglut_2018_01_14.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} BB182187285E06880F0EDE3A39530091)
475
  endif (WITH_3D_RENDERER)
476 477 478 479 480 481 482 483
  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)
484

485 486 487
  # If OpenPose builds it
  if (BUILD_CAFFE)
    unset(Caffe_INCLUDE_DIRS CACHE)
488 489
    unset(Caffe_LIB CACHE)
    unset(Caffe_Proto_LIB CACHE)
490 491 492 493
  endif (BUILD_CAFFE)
  # OpenCL
  if (${GPU_MODE} MATCHES "OPENCL")
    set(VCXPROJ_FILE_GPU_MODE "_CL")
494 495
    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)
496
  # CPU & CUDA
497
  else (${GPU_MODE} MATCHES "OPENCL")
498 499 500 501 502 503 504 505 506 507 508
    # 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")
509
  endif (${GPU_MODE} MATCHES "OPENCL")
R
Raaj 已提交
510 511 512 513 514 515 516 517 518 519 520 521
  # 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 ()
522
  if (WITH_3D_RENDERER)
523 524
    find_library(GLUT_LIBRARY freeglut HINTS ${FIND_LIB_PREFIX}/freeglut/lib)
    message(STATUS "\${GLUT_LIBRARY} = ${GLUT_LIBRARY}")
525 526 527 528
  endif (WITH_3D_RENDERER)
  if (WITH_FLIR_CAMERA)
    find_library(SPINNAKER_LIB spinnaker_v140 HINTS ${FIND_LIB_PREFIX}/spinnaker/lib)
  endif (WITH_FLIR_CAMERA)
529 530
  set(Caffe_LIBS ${Caffe_LIB};${Caffe_Proto_LIB})
  set(OpenCV_INCLUDE_DIRS "3rdparty/windows/opencv/include")
531
  # OpenCL
532 533 534
  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")
535
  # CUDA and CPU
536
  else (${GPU_MODE} MATCHES "OPENCL")
537 538 539 540 541 542 543
    # 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")
544
  endif (${GPU_MODE} MATCHES "OPENCL")
545 546
  set(Boost_INCLUDE_DIRS "3rdparty/windows/caffe3rdparty/include/boost-1_61")
  set(WINDOWS_INCLUDE_DIRS "3rdparty/windows/caffe3rdparty/include")
547
  if (WITH_3D_RENDERER)
548
    set(GLUT_INCLUDE_DIRS "3rdparty/windows/freeglut/include")
549 550 551 552
  endif (WITH_3D_RENDERER)
  if (WITH_FLIR_CAMERA)
    set(SPINNAKER_INCLUDE_DIRS "3rdparty/windows/spinnaker/include")
  endif (WITH_FLIR_CAMERA)
B
Bikramjot Hanzra 已提交
553
  set(Caffe_FOUND 1)
R
Raaj 已提交
554 555 556

  # Build DLL Must be on if Build Python is on
  if (BUILD_PYTHON)
G
gineshidalgo99 已提交
557 558
    if (NOT BUILD_BIN_FOLDER)
      message(FATAL_ERROR "BUILD_BIN_FOLDER must be turned on to as well to build python library")
G
gineshidalgo99 已提交
559 560
    endif (NOT BUILD_BIN_FOLDER)
  endif (BUILD_PYTHON)
R
Raaj 已提交
561 562

  # Auto copy DLLs
G
gineshidalgo99 已提交
563 564 565
  if (BUILD_BIN_FOLDER)
    # Locate DLLs
    # Caffe DLLs
R
Raaj 已提交
566 567 568 569 570 571 572
    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 已提交
573
    # Caffe 3rd-party DLLs
574
    file(GLOB CAFFE_3RD_PARTY_DLL "${CMAKE_SOURCE_DIR}/3rdparty/windows/caffe3rdparty/lib/*.dll")
G
gineshidalgo99 已提交
575 576
    # OpenCV DLLs
    file(GLOB OPENCV_DLL "${CMAKE_SOURCE_DIR}/3rdparty/windows/opencv/x64/vc14/bin/*.dll")
577 578 579 580
    # 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 已提交
581 582 583 584 585
    # 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})
586 587 588
    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 已提交
589
  endif (BUILD_BIN_FOLDER)
R
Raaj 已提交
590

591
endif (WIN32)
B
Bikramjot Hanzra 已提交
592 593


G
gineshidalgo99 已提交
594
### CAFFE
595
if (UNIX OR APPLE)
B
Bikramjot Hanzra 已提交
596

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

599 600 601 602 603 604
    # 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 已提交
605

606 607
    # Else build from scratch
    if (BUILD_CAFFE)
B
Bikramjot Hanzra 已提交
608

609 610
      # Download Caffe
      message(STATUS "Caffe will be downloaded from source now. NOTE: This process might take several minutes depending
G
gineshidalgo99 已提交
611 612 613 614 615 616 617
        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)
618
      else (CAFFE_DIR_VALID_LENGTH EQUAL 0)
G
gineshidalgo99 已提交
619
        message(STATUS "Caffe has already been downloaded.")
620
      endif (CAFFE_DIR_VALID_LENGTH EQUAL 0)
G
gineshidalgo99 已提交
621 622 623 624 625 626 627

      # 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 已提交
628 629
          #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 已提交
630 631
          execute_process(COMMAND sh prepare_mkl.sh WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/3rdparty/caffe/external/mkl
              OUTPUT_VARIABLE rv)
G
gineshidalgo99 已提交
632 633 634
          set( MLIST ${rv} )
          separate_arguments(MLIST)
          list(GET MLIST 0 MKL_PATH)
635
          message(STATUS ${MKL_PATH})
G
gineshidalgo99 已提交
636 637 638 639 640
          file(GLOB MKL_SO
            "${MKL_PATH}lib/*"
            )
          file(COPY ${MKL_SO} DESTINATION ${CMAKE_BINARY_DIR}/caffe)

G
gineshidalgo99 已提交
641 642 643 644 645 646 647 648 649 650 651 652
          # 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 已提交
653
          set(MKL_LIBS
G
gineshidalgo99 已提交
654
            #"${CMAKE_BINARY_DIR}/caffe/libmlsl.so"
G
gineshidalgo99 已提交
655 656 657 658
            "${CMAKE_BINARY_DIR}/caffe/libiomp5.so"
            "${CMAKE_BINARY_DIR}/caffe/libmklml_intel.so"
            "${CMAKE_BINARY_DIR}/caffe/lib/libmkldnn.so"
            )
659
        else (USE_MKL)
G
gineshidalgo99 已提交
660
          execute_process(COMMAND git checkout master WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
661
        endif (USE_MKL)
G
gineshidalgo99 已提交
662 663
        set(CAFFE_CPU_ONLY ON)
        set(USE_CUDNN OFF)
664 665 666
      elseif (${GPU_MODE} MATCHES "OPENCL")
        execute_process(COMMAND git checkout opencl WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
        set(USE_CUDNN OFF)
667
      endif (${GPU_MODE} MATCHES "CUDA")
G
gineshidalgo99 已提交
668

669 670
      # Build Caffe
      message(STATUS "Caffe will be built from source now.")
B
Bikramjot Hanzra 已提交
671
      find_package(Caffe)
672 673 674
      include(ExternalProject)
      set(CAFFE_PREFIX caffe)
      set(CAFFE_URL ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
B
Bikramjot Hanzra 已提交
675

G
gineshidalgo99 已提交
676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694
      # 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})
695
      else (USE_MKL)
G
gineshidalgo99 已提交
696
        ExternalProject_Add(openpose_caffe
697 698
          SOURCE_DIR ${CAFFE_URL}
          PREFIX ${CAFFE_PREFIX}
G
gineshidalgo99 已提交
699
          CMAKE_ARGS -DCMAKE_INSTALL_PREFIX:PATH=<INSTALL_DIR>
G
gineshidalgo99 已提交
700 701 702 703 704 705 706 707 708 709
          -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})
710
      endif (USE_MKL)
B
Bikramjot Hanzra 已提交
711

712 713
      ExternalProject_Get_Property(openpose_caffe install_dir)

B
Bikramjot Hanzra 已提交
714 715 716
      if (NOT Caffe_FOUND)
        add_custom_command(TARGET openpose_caffe
            POST_BUILD
717
            COMMAND ${CMAKE_COMMAND} ${CMAKE_SOURCE_DIR}
B
Bikramjot Hanzra 已提交
718 719 720 721
            COMMAND $(MAKE)
            WORKING_DIRECTORY ${CMAKE_BINARY_DIR}
            COMMENT "Rerunning cmake after building Caffe submodule")
      endif (NOT Caffe_FOUND)
722 723 724 725

    endif (BUILD_CAFFE)

    if (NOT Caffe_FOUND AND NOT BUILD_CAFFE)
G
gineshidalgo99 已提交
726
      message(FATAL_ERROR "Caffe not found. Either turn on the BUILD_CAFFE option or specify the path of Caffe includes
727
        and libs using -DCaffe_INCLUDE_DIRS and -DCaffe_LIBS.")
728 729 730 731
    endif (NOT Caffe_FOUND AND NOT BUILD_CAFFE)

  endif (${DL_FRAMEWORK} MATCHES "CAFFE")

732
endif (UNIX OR APPLE)
B
Bikramjot Hanzra 已提交
733 734 735 736

### PROJECT INCLUDES
# Specify the include directories
include_directories(
737
  include
B
Bikramjot Hanzra 已提交
738 739 740
  ${GFLAGS_INCLUDE_DIR}
  ${GLOG_INCLUDE_DIR}
  ${OpenCV_INCLUDE_DIRS})
B
Bikramjot Hanzra 已提交
741

742 743 744 745 746 747 748 749
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)

750 751 752 753 754 755
# Calibration
if (NOT ${WITH_EIGEN} MATCHES "NONE")
  include_directories(
      ${EIGEN3_INCLUDE_DIRS})
endif (NOT ${WITH_EIGEN} MATCHES "NONE")

756 757 758 759 760
if (APPLE)
  include_directories(
      "/usr/local/opt/openblas/include")
endif (APPLE)

761
if (USE_MKL)
762 763 764 765
 include_directories(
     "${MKL_PATH}/include/")
endif (USE_MKL)

B
Bikramjot Hanzra 已提交
766 767 768 769 770
if (Caffe_FOUND)
  include_directories(
      ${Caffe_INCLUDE_DIRS})
endif (Caffe_FOUND)

G
gineshidalgo99 已提交
771 772
if (${GPU_MODE} MATCHES "CUDA")
  include_directories(
B
Bikramjot Hanzra 已提交
773
      ${CUDA_INCLUDE_DIRS})
774
elseif (${GPU_MODE} MATCHES "OPENCL")
775 776
  include_directories(
    ${OpenCL_INCLUDE_DIRS})
777
endif (${GPU_MODE} MATCHES "CUDA")
778
# 3D
779 780 781
if (WITH_3D_RENDERER)
  include_directories(${GLUT_INCLUDE_DIRS})
endif (WITH_3D_RENDERER)
782 783 784
if (WITH_CERES)
  include_directories(${CERES_INCLUDE_DIRS})
endif (WITH_CERES)
785 786 787
if (WITH_FLIR_CAMERA)
  include_directories(${SPINNAKER_INCLUDE_DIRS})
endif (WITH_FLIR_CAMERA)
788 789 790 791 792 793 794 795
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)
796
# Windows includes
797 798 799 800 801
if (WIN32)
  include_directories(
    ${Boost_INCLUDE_DIRS}
    ${WINDOWS_INCLUDE_DIRS})
endif (WIN32)
B
Bikramjot Hanzra 已提交
802 803


804 805
### COLLECT ALL 3RD-PARTY LIBRARIES TO BE LINKED AGAINST
set(OpenPose_3rdparty_libraries ${OpenCV_LIBS} ${GLOG_LIBRARY})
806
if (UNIX OR APPLE)
807 808 809 810 811
  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})
812
endif (UNIX OR APPLE)
813 814 815 816 817 818 819 820 821 822 823
# 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 已提交
824 825 826 827 828 829 830 831 832
# 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)
833
# 3-D
834 835 836 837
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)
838 839 840 841 842 843 844 845 846
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 已提交
847
# Pthread
848
if (UNIX OR APPLE)
G
gineshidalgo99 已提交
849
  set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries} pthread)
850
endif (UNIX OR APPLE)
851 852 853 854

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


B
Bikramjot Hanzra 已提交
855 856 857 858 859 860 861
### ADD SUBDIRECTORIES
if (Caffe_FOUND)
  add_subdirectory(src)
  if (BUILD_EXAMPLES)
    add_subdirectory(examples)
  endif (BUILD_EXAMPLES)
endif (Caffe_FOUND)
B
Bikramjot Hanzra 已提交
862 863


G
gineshidalgo99 已提交
864
### DOWNLOAD MODELS
B
Bikramjot Hanzra 已提交
865 866 867 868 869
# 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 已提交
870

871 872 873
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 已提交
874
  5156d31f670511fce9b4e28b403f2939) # Body (COCO)
875
download_model("body (MPI)" ${DOWNLOAD_BODY_MPI_MODEL} pose/mpi/pose_iter_160000.caffemodel
G
gineshidalgo99 已提交
876
  2ca0990c7562bd7ae03f3f54afa96e00) # Body (MPI)
877
download_model("face" ${DOWNLOAD_FACE_MODEL} face/pose_iter_116000.caffemodel
G
gineshidalgo99 已提交
878
  e747180d728fa4e4418c465828384333) # Face
879
download_model("hand" ${DOWNLOAD_HAND_MODEL} hand/pose_iter_102000.caffemodel
G
gineshidalgo99 已提交
880
  a82cfc3fea7c62f159e11bd3674c1531) # Hand
B
Bikramjot Hanzra 已提交
881 882

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

R
Raaj 已提交
884 885 886 887
### PYTHON
if (Caffe_FOUND)
  if(BUILD_PYTHON)
    add_subdirectory(python)
G
gineshidalgo99 已提交
888 889
  endif (BUILD_PYTHON)
endif(Caffe_FOUND)
890

G
gineshidalgo99 已提交
891
### GENERATE DOCUMENTATION
892
if (UNIX OR APPLE)
893 894 895 896 897 898 899 900 901

  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 已提交
902 903 904 905
        COMMAND ${DOXYGEN_EXECUTABLE} ${DOXYGEN_FILE}
        WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/doc
        COMMENT "Generating API documentation with Doxygen"
        VERBATIM)
906
    else (DOXYGEN_FOUND)
907
      message(FATAL_ERROR "Doxygen needs to be installed to generate the doxygen documentation.")
908 909 910
    endif (DOXYGEN_FOUND)
  endif (BUILD_DOCS)

911
endif (UNIX OR APPLE)
912

B
bikz05 已提交
913

G
gineshidalgo99 已提交
914
### INSTALL
915
if (UNIX OR APPLE)
B
Bikramjot Hanzra 已提交
916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941
  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)
942
endif (UNIX OR APPLE)