CMakeLists.txt 37.0 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


G
gineshidalgo99 已提交
29
### CMAKE_BUILD_TYPE
30
# Default: Release
G
gineshidalgo99 已提交
31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50
# Bug fixed: By default, it uses something different to Release, that provokes OpenPose to be about 15% slower than
# it should be.
# Is CMAKE_BUILD_TYPE "Debug" or "MinSizeRel" or "RelWithDebInfo"?
set(CMAKE_BUILD_TYPE_KNOWN FALSE)
if (${CMAKE_BUILD_TYPE} MATCHES "Debug")
  set(CMAKE_BUILD_TYPE_KNOWN TRUE)
endif (${CMAKE_BUILD_TYPE} MATCHES "Debug")
if (${CMAKE_BUILD_TYPE} MATCHES "MinSizeRel")
  set(CMAKE_BUILD_TYPE_KNOWN TRUE)
endif (${CMAKE_BUILD_TYPE} MATCHES "MinSizeRel")
if (${CMAKE_BUILD_TYPE} MATCHES "RelWithDebInfo")
  set(CMAKE_BUILD_TYPE_KNOWN TRUE)
endif (${CMAKE_BUILD_TYPE} MATCHES "RelWithDebInfo")
# Assign proper CMAKE_BUILD_TYPE
if (${CMAKE_BUILD_TYPE_KNOWN})
  set(CMAKE_BUILD_TYPE "Release" CACHE STRING "Choose the type of build.")
else (${CMAKE_BUILD_TYPE_KNOWN})
  set(CMAKE_BUILD_TYPE "Release" CACHE STRING "Choose the type of build." FORCE)
endif (${CMAKE_BUILD_TYPE_KNOWN})
set_property(CACHE CMAKE_BUILD_TYPE PROPERTY STRINGS "Release" "Debug" "MinSizeRel" "RelWithDebInfo")
51 52


G
gineshidalgo99 已提交
53
### FLAGS
54
if (WIN32)
G
gineshidalgo99 已提交
55
  # TODO -- Makeshift solution -- This prevents rerunning build again
56 57 58
  # https://gitlab.kitware.com/cmake/cmake/issues/16783
  set(CMAKE_SUPPRESS_REGENERATION ON)

A
arkceajin 已提交
59 60
  string (REPLACE "/D_WINDOWS" "" CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}")
  string (REPLACE "/DWIN32" "" CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}")
61 62 63 64

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

G
gineshidalgo99 已提交
65
  set(CMAKE_CONFIGURATION_TYPES Release Debug CACHE TYPE INTERNAL FORCE)
66

67 68 69 70 71 72 73 74 75
  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")
76

77 78 79
  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}")
80
elseif (UNIX)
G
gineshidalgo99 已提交
81
  # Turn on C++11
82 83
  add_definitions(-std=c++11)
  set(CMAKE_CXX_FLAGS_RELEASE "-O3")
84 85 86 87
elseif (APPLE)
  # Turn on C++11
  set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11")
  set(CMAKE_CXX_FLAGS_RELEASE "-O3")
88
endif (WIN32)
B
Bikramjot Hanzra 已提交
89 90

# C++ additional flags
G
gineshidalgo99 已提交
91
if (CMAKE_COMPILER_IS_GNUCXX)
92 93 94 95 96 97 98 99

  # 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")
100
  endif (${INSTRUCTION_SET} MATCHES "SSE4")
101 102 103 104

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

G
gineshidalgo99 已提交
107
  message(STATUS "GCC detected, adding compile flags")
108
  set(OP_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${SIMD_FLAGS} -fopenmp -Wpedantic -Wall -Wextra -Wfatal-errors")
G
gineshidalgo99 已提交
109
endif (CMAKE_COMPILER_IS_GNUCXX)
B
Bikramjot Hanzra 已提交
110 111


G
gineshidalgo99 已提交
112
### PROJECT OPTIONS
B
Bikramjot Hanzra 已提交
113 114 115 116 117 118 119 120 121
# 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 已提交
122
    "DL_FRAMEWORK" ON)
123 124
  # OpenPose flags
  add_definitions(-DUSE_CAFFE)
B
Bikramjot Hanzra 已提交
125 126
endif (${DL_FRAMEWORK} MATCHES "CAFFE")

127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145
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 已提交
146
# Set the acceleration library
G
gineshidalgo99 已提交
147 148 149 150 151
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)
152 153 154 155
# 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 已提交
156
#   set_property(CACHE GPU_MODE PROPERTY STRINGS CUDA CPU_ONLY)
157 158 159 160
# elseif (OpenCL_FOUND)
#   set_property(CACHE GPU_MODE PROPERTY STRINGS OPENCL CPU_ONLY)
# else ()
#   set_property(CACHE GPU_MODE PROPERTY STRINGS CPU_ONLY)
161
# endif (CUDA_FOUND AND OpenCL_FOUND)
R
Raaj 已提交
162
set_property(CACHE GPU_MODE PROPERTY STRINGS CUDA OPENCL CPU_ONLY)
G
gineshidalgo99 已提交
163

164
# Look for CUDA
G
gineshidalgo99 已提交
165
set(CUDA_FOUND FALSE)
166 167 168 169 170
if (${GPU_MODE} MATCHES "CUDA")
  find_package(CUDA)
endif (${GPU_MODE} MATCHES "CUDA")
# Look for OpenCL
set(OpenCL_FOUND FALSE)
G
gineshidalgo99 已提交
171
set(CUDA_VERSION_MAJOR 0)
172 173 174 175
if (${GPU_MODE} MATCHES "OPENCL")
  find_package(OpenCL)
endif (${GPU_MODE} MATCHES "OPENCL")

176 177 178 179 180 181 182
# 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)
183
endif (CUDA_FOUND AND OpenCL_FOUND AND ${CUDA_VERSION_MAJOR} LESS 9)
184
# Handle desired GPU mode option
185 186 187
if (${GPU_MODE} MATCHES "CUDA")
  # OpenPose flags
  add_definitions(-DUSE_CUDA)
G
gineshidalgo99 已提交
188
  message(STATUS "Building with CUDA.")
189
elseif (${GPU_MODE} MATCHES "CPU_ONLY")
G
gineshidalgo99 已提交
190 191
  add_definitions(-DUSE_CPU_ONLY)
  message(STATUS "Building CPU Only.")
192 193
  # OpenPose flag for Caffe
  add_definitions(-DCPU_ONLY)
194 195 196 197 198
elseif (${GPU_MODE} MATCHES "OPENCL")
  # OpenPose flag for Caffe
  add_definitions(-DUSE_OPENCL)
  add_definitions(-DUSE_GREENTEA)
  message(STATUS "Building with OpenCL.")
199
endif (${GPU_MODE} MATCHES "CUDA")
G
gineshidalgo99 已提交
200 201 202 203 204

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

208 209 210 211 212 213
if (${USE_MKL})
  # OpenPose flags
  add_definitions(-DUSE_MKL)
  message(STATUS "Building with MKL support.")
endif (${USE_MKL})

214 215 216 217 218
# Set/disable profiler
if (PROFILER_ENABLED)
  add_definitions(-DPROFILER_ENABLED)
endif (PROFILER_ENABLED)

B
Bikramjot Hanzra 已提交
219 220 221 222 223 224 225 226 227 228
# 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")
229

G
gineshidalgo99 已提交
230
# Suboptions for OpenPose 3D Reconstruction module and demo
231
option(WITH_3D_RENDERER "Add OpenPose 3D renderer module (it requires FreeGLUT library)." OFF)
232
if (UNIX AND NOT APPLE)
G
gineshidalgo99 已提交
233
  option(WITH_CERES "Add Ceres support for advanced 3-D reconstruction." OFF)
234
endif (UNIX AND NOT APPLE)
235
option(WITH_FLIR_CAMERA "Add FLIR (formerly Point Grey) camera code (requires Spinnaker SDK already installed)." OFF)
G
gineshidalgo99 已提交
236
# option(WITH_3D_ADAM_MODEL "Add 3-D Adam model (requires OpenGL, Ceres, Eigen, OpenMP, FreeImage, GLEW, and IGL already installed)." OFF)
237

G
gineshidalgo99 已提交
238
# Faster GUI rendering
239 240 241
# 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 已提交
242

243
# Set the acceleration library
G
gineshidalgo99 已提交
244
if (WIN32 OR APPLE)
245 246 247 248 249
  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 已提交
250
endif (WIN32 OR APPLE)
251

G
gineshidalgo99 已提交
252 253 254 255 256
# # 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 已提交
257
# Download the models
258 259 260
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 已提交
261
option(DOWNLOAD_FACE_MODEL "Download face model." ON)
262 263 264
option(DOWNLOAD_HAND_MODEL "Download hand model." ON)

# Asio
G
gineshidalgo99 已提交
265
# option(USE_ASIO "Include Asio header-only library." OFF)
B
Bikramjot Hanzra 已提交
266 267 268 269

# More options
option(BUILD_EXAMPLES "Build OpenPose examples." ON)
option(BUILD_DOCS "Build OpenPose documentation." OFF)
R
Raaj 已提交
270
option(BUILD_PYTHON "Build OpenPose python." OFF)
R
Raaj 已提交
271
if (WIN32)
G
gineshidalgo99 已提交
272
  option(BUILD_BIN_FOLDER "Copy all required 3rd-party DLL files into {build_directory}/bin. Disable to save some memory." ON)
G
gineshidalgo99 已提交
273 274 275
endif (WIN32)

# Unity
T
Tianyi Zhao 已提交
276
option(BUILD_UNITY_SUPPORT "Build OpenPose as a Unity plugin." OFF)
B
Bikramjot Hanzra 已提交
277

B
bikz05 已提交
278
# Build as shared library
R
Raaj 已提交
279
option(BUILD_SHARED_LIBS "Build as shared lib." ON)
B
bikz05 已提交
280

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

G
gineshidalgo99 已提交
284
# Threads - Pthread
285
if (${GPU_MODE} MATCHES "OPENCL" OR (UNIX OR APPLE))
286 287
  unset(CMAKE_THREAD_LIBS_INIT CACHE)
  find_package(Threads)
288 289 290 291
endif (${GPU_MODE} MATCHES "OPENCL" OR (UNIX OR APPLE))

# Caffe OpenCL Boost Issue
if (APPLE)
R
Raaj 已提交
292
  if (${GPU_MODE} MATCHES "OPENCL" OR BUILD_PYTHON)
293 294 295 296 297
    unset(Boost_SYSTEM_LIBRARY CACHE)
    find_package(Boost COMPONENTS system REQUIRED)
  else ()
    set(Boost_SYSTEM_LIBRARY "")
  endif ()
G
gineshidalgo99 已提交
298
endif (APPLE)
B
Bikramjot Hanzra 已提交
299

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

G
gineshidalgo99 已提交
303 304
if (${GPU_MODE} MATCHES "CUDA")
  find_package(CUDA)
305
endif (${GPU_MODE} MATCHES "CUDA")
G
gineshidalgo99 已提交
306

307
# Adding 3D
G
gineshidalgo99 已提交
308 309 310 311
if (WITH_OPENCV_WITH_OPENGL)
  # OpenPose flags
  add_definitions(-DUSE_OPENCV_WITH_OPENGL)
endif (WITH_OPENCV_WITH_OPENGL)
312
if (WITH_3D_RENDERER)
313
  # OpenPose flags
314
  add_definitions(-DUSE_3D_RENDERER)
315
endif (WITH_3D_RENDERER)
316
if (WITH_CERES)
317
  add_definitions(-DUSE_CERES)
318
endif (WITH_CERES)
319 320
if (WITH_FLIR_CAMERA)
  # OpenPose flags
321
  add_definitions(-DUSE_FLIR_CAMERA)
322
endif (WITH_FLIR_CAMERA)
323 324 325 326
if (WITH_3D_ADAM_MODEL)
  # OpenPose flags
  add_definitions(-DUSE_3D_ADAM_MODEL)
endif (WITH_3D_ADAM_MODEL)
327

G
gineshidalgo99 已提交
328 329 330
# Adding tracking
if (WITH_TRACKING)
  # OpenPose flags
331
  add_definitions(-DUSE_TRACKING)
G
gineshidalgo99 已提交
332 333
endif (WITH_TRACKING)

G
gineshidalgo99 已提交
334 335 336 337 338 339
# Unity
if (BUILD_UNITY_SUPPORT)
  # OpenPose flags
  add_definitions(-DUSE_UNITY_SUPPORT)
endif (BUILD_UNITY_SUPPORT)

340 341 342 343 344 345 346 347 348
# 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
349
  add_definitions(-DUSE_EIGEN)
350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365
  # 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")

366
if (UNIX OR APPLE)
G
gineshidalgo99 已提交
367 368 369
  if (${GPU_MODE} MATCHES "CUDA")
    include(cmake/Cuda.cmake)
    find_package(CuDNN)
370
  endif (${GPU_MODE} MATCHES "CUDA")
J
Jim Carroll 已提交
371 372 373 374
  find_package(GFlags) # For Caffe and OpenPose
  find_package(Glog) # For Caffe
  find_package(Protobuf REQUIRED) # For Caffe

B
Bikramjot Hanzra 已提交
375 376
  if (OpenCV_CONFIG_FILE)
    include (${OpenCV_CONFIG_FILE})
J
Jim Carroll 已提交
377 378 379
  # Allow explicitly setting the OpenCV includes and libs
  elseif (OpenCV_INCLUDE_DIRS AND OpenCV_LIBS)
    set(OpenCV_FOUND 1)
B
Bikramjot Hanzra 已提交
380 381 382
  elseif (OpenCV_INCLUDE_DIRS AND OpenCV_LIBS_DIR)
    file(GLOB_RECURSE OpenCV_LIBS "${OpenCV_LIBS_DIR}*.so")
    set(OpenCV_FOUND 1)
383
  else (OpenCV_CONFIG_FILE)
B
Bikramjot Hanzra 已提交
384 385
    find_package(OpenCV)
  endif (OpenCV_CONFIG_FILE)
386

387
  # 3D
388
  if (WITH_3D_RENDERER)
389 390
    # GLUT
    find_package(GLUT REQUIRED)
391 392
    # OpenGL
    find_package(OpenGL REQUIRED)
393
  endif (WITH_3D_RENDERER)
394 395 396 397
  if (WITH_CERES)
    # Eigen + Ceres
    find_package(Ceres REQUIRED COMPONENTS SuiteSparse)
  endif (WITH_CERES)
398
  if (WITH_FLIR_CAMERA)
399 400 401
    # Spinnaker
    find_package(Spinnaker)
    if (NOT SPINNAKER_FOUND)
402 403
      message(FATAL_ERROR "Spinnaker not found. Either turn off the `WITH_FLIR_CAMERA` option or specify the path to
        the Spinnaker includes and libs.")
404
    endif (NOT SPINNAKER_FOUND)
405
  endif (WITH_FLIR_CAMERA)
406
  if (WITH_3D_ADAM_MODEL)
G
gineshidalgo99 已提交
407 408
    if (NOT WITH_3D_RENDERER)
      message(FATAL_ERROR "WITH_3D_RENDERER is required if WITH_3D_ADAM_MODEL is enabled.")
409
    endif (NOT WITH_3D_RENDERER)
410 411
    find_package(PkgConfig)
    pkg_check_modules(EIGEN3 REQUIRED eigen3)
G
gineshidalgo99 已提交
412 413
    # Others: sudo apt-get install libglm-dev
    # http://ceres-solver.org
414
    find_package(Ceres REQUIRED COMPONENTS SuiteSparse)
G
gineshidalgo99 已提交
415
    # sudo apt-get install libglew-dev
416
    find_package(GLEW REQUIRED)
G
gineshidalgo99 已提交
417 418 419 420
    # 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
421 422
    find_package(LIBIGL REQUIRED)
    find_package(OpenMP REQUIRED)
G
gineshidalgo99 已提交
423 424
    # Only adam/renderer::Renderer::IdleSaveImage() uses it. Make dependency optional in hand_model
    # FIND_LIBRARY(FREE_IMAGE_LIBRARY NAMES libfreeimage.so)
425 426 427 428
    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)
429

G
gineshidalgo99 已提交
430 431 432 433
  # OpenMP
  if (${GPU_MODE} MATCHES "CPU_ONLY")
    find_package(OpenMP)
    if (OPENMP_FOUND)
G
gineshidalgo99 已提交
434 435 436
      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}")
437 438
    endif (OPENMP_FOUND)
  endif (${GPU_MODE} MATCHES "CPU_ONLY")
G
gineshidalgo99 已提交
439 440 441 442 443 444 445

  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.")
446
      execute_process(COMMAND cat install_cuda.sh WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/scripts/ubuntu)
G
gineshidalgo99 已提交
447 448 449 450 451
      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.")
452
      execute_process(COMMAND cat install_cudnn.sh WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/scripts/ubuntu)
G
gineshidalgo99 已提交
453 454
      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)
455
  endif (${GPU_MODE} MATCHES "CUDA")
456 457

  if (NOT GLOG_FOUND)
G
gineshidalgo99 已提交
458 459
    message(FATAL_ERROR "Glog not found. Install Glog from the command line using the command(s) -\
      sudo apt-get install libgoogle-glog-dev")
460 461 462
  endif (NOT GLOG_FOUND)

  if (NOT GFLAGS_FOUND)
G
gineshidalgo99 已提交
463
    message(FATAL_ERROR "GFlags not found. Install GFlags from the command line using the command(s) --\
464 465 466 467 468 469 470 471
      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)

472
endif (UNIX OR APPLE)
473 474 475

if (WIN32)

G
gineshidalgo99 已提交
476
  if ("${CMAKE_VERSION}" VERSION_GREATER 3.6.3)
477
    set_property(DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR} PROPERTY VS_STARTUP_PROJECT OpenPoseDemo)
G
gineshidalgo99 已提交
478
  endif ("${CMAKE_VERSION}" VERSION_GREATER 3.6.3)
479 480 481 482 483 484 485 486 487

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

488
  # Download required zip files
489
  download_zip("opencv_310.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} 1e5240a64b814b3c0b822f136be78ad7)
G
gineshidalgo99 已提交
490
  download_zip("caffe3rdparty_2018_06_26.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} 892A39C0CFBAA11CA8648B125852E01F)
491 492
  if (${GPU_MODE} MATCHES "OPENCL")
    download_zip("caffe_opencl_2018_02_13.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} 3ac3e1acf5ee6a4e57920be73053067a)
493 494
  elseif (${GPU_MODE} MATCHES "CPU_ONLY")
    download_zip("caffe_cpu_2018_05_27.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} 87E8401B6DFBAC5B8E909DD20E3B3390)
495
  else (${GPU_MODE} MATCHES "OPENCL")
496
    download_zip("caffe_2018_01_18.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} 4b8e548cc7ea20abea472950dd5301bd)
497
  endif (${GPU_MODE} MATCHES "OPENCL")
498
  if (WITH_3D_RENDERER)
499
    download_zip("freeglut_2018_01_14.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} BB182187285E06880F0EDE3A39530091)
500
  endif (WITH_3D_RENDERER)
501 502 503 504 505 506 507 508
  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)
509

510 511 512
  # If OpenPose builds it
  if (BUILD_CAFFE)
    unset(Caffe_INCLUDE_DIRS CACHE)
513 514
    unset(Caffe_LIB CACHE)
    unset(Caffe_Proto_LIB CACHE)
515 516 517 518
  endif (BUILD_CAFFE)
  # OpenCL
  if (${GPU_MODE} MATCHES "OPENCL")
    set(VCXPROJ_FILE_GPU_MODE "_CL")
519 520
    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)
521
  # CPU & CUDA
522
  else (${GPU_MODE} MATCHES "OPENCL")
523 524 525 526 527 528 529 530 531 532 533
    # 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")
534
  endif (${GPU_MODE} MATCHES "OPENCL")
R
Raaj 已提交
535 536 537 538 539 540 541 542 543 544 545 546
  # 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 ()
547
  if (WITH_3D_RENDERER)
548 549
    find_library(GLUT_LIBRARY freeglut HINTS ${FIND_LIB_PREFIX}/freeglut/lib)
    message(STATUS "\${GLUT_LIBRARY} = ${GLUT_LIBRARY}")
550 551 552 553
  endif (WITH_3D_RENDERER)
  if (WITH_FLIR_CAMERA)
    find_library(SPINNAKER_LIB spinnaker_v140 HINTS ${FIND_LIB_PREFIX}/spinnaker/lib)
  endif (WITH_FLIR_CAMERA)
554 555
  set(Caffe_LIBS ${Caffe_LIB};${Caffe_Proto_LIB})
  set(OpenCV_INCLUDE_DIRS "3rdparty/windows/opencv/include")
556
  # OpenCL
557 558 559
  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")
560
  # CUDA and CPU
561
  else (${GPU_MODE} MATCHES "OPENCL")
562 563 564 565 566 567 568
    # 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")
569
  endif (${GPU_MODE} MATCHES "OPENCL")
570 571
  set(Boost_INCLUDE_DIRS "3rdparty/windows/caffe3rdparty/include/boost-1_61")
  set(WINDOWS_INCLUDE_DIRS "3rdparty/windows/caffe3rdparty/include")
572
  if (WITH_3D_RENDERER)
573
    set(GLUT_INCLUDE_DIRS "3rdparty/windows/freeglut/include")
574 575 576 577
  endif (WITH_3D_RENDERER)
  if (WITH_FLIR_CAMERA)
    set(SPINNAKER_INCLUDE_DIRS "3rdparty/windows/spinnaker/include")
  endif (WITH_FLIR_CAMERA)
B
Bikramjot Hanzra 已提交
578
  set(Caffe_FOUND 1)
R
Raaj 已提交
579 580 581

  # Build DLL Must be on if Build Python is on
  if (BUILD_PYTHON)
G
gineshidalgo99 已提交
582 583
    if (NOT BUILD_BIN_FOLDER)
      message(FATAL_ERROR "BUILD_BIN_FOLDER must be turned on to as well to build python library")
G
gineshidalgo99 已提交
584 585
    endif (NOT BUILD_BIN_FOLDER)
  endif (BUILD_PYTHON)
R
Raaj 已提交
586 587

  # Auto copy DLLs
G
gineshidalgo99 已提交
588 589 590
  if (BUILD_BIN_FOLDER)
    # Locate DLLs
    # Caffe DLLs
R
Raaj 已提交
591 592 593 594 595 596 597
    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 已提交
598
    # Caffe 3rd-party DLLs
599
    file(GLOB CAFFE_3RD_PARTY_DLL "${CMAKE_SOURCE_DIR}/3rdparty/windows/caffe3rdparty/lib/*.dll")
G
gineshidalgo99 已提交
600 601
    # OpenCV DLLs
    file(GLOB OPENCV_DLL "${CMAKE_SOURCE_DIR}/3rdparty/windows/opencv/x64/vc14/bin/*.dll")
602 603 604 605
    # 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 已提交
606 607 608 609 610
    # 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})
611 612 613
    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 已提交
614
  endif (BUILD_BIN_FOLDER)
R
Raaj 已提交
615

616
endif (WIN32)
B
Bikramjot Hanzra 已提交
617 618


G
gineshidalgo99 已提交
619
### CAFFE
620
if (UNIX OR APPLE)
B
Bikramjot Hanzra 已提交
621

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

624 625 626 627 628 629
    # 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 已提交
630

631 632
    # Else build from scratch
    if (BUILD_CAFFE)
B
Bikramjot Hanzra 已提交
633

634 635
      # Download Caffe
      message(STATUS "Caffe will be downloaded from source now. NOTE: This process might take several minutes depending
G
gineshidalgo99 已提交
636 637 638 639 640 641
        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)
R
Raaj 已提交
642
        execute_process(COMMAND git submodule update --init ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
643
        # execute_process(COMMAND git submodule update --init --recursive --remote) # This would initialize them all
644
      else (CAFFE_DIR_VALID_LENGTH EQUAL 0)
G
gineshidalgo99 已提交
645
        message(STATUS "Caffe has already been downloaded.")
646
      endif (CAFFE_DIR_VALID_LENGTH EQUAL 0)
G
gineshidalgo99 已提交
647 648 649 650 651 652 653

      # 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 已提交
654 655
          #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 已提交
656 657
          execute_process(COMMAND sh prepare_mkl.sh WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/3rdparty/caffe/external/mkl
              OUTPUT_VARIABLE rv)
G
gineshidalgo99 已提交
658 659 660
          set( MLIST ${rv} )
          separate_arguments(MLIST)
          list(GET MLIST 0 MKL_PATH)
661
          message(STATUS ${MKL_PATH})
G
gineshidalgo99 已提交
662 663 664 665 666
          file(GLOB MKL_SO
            "${MKL_PATH}lib/*"
            )
          file(COPY ${MKL_SO} DESTINATION ${CMAKE_BINARY_DIR}/caffe)

G
gineshidalgo99 已提交
667 668 669 670 671 672 673 674 675 676 677 678
          # 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 已提交
679
          set(MKL_LIBS
G
gineshidalgo99 已提交
680
            #"${CMAKE_BINARY_DIR}/caffe/libmlsl.so"
G
gineshidalgo99 已提交
681 682 683 684
            "${CMAKE_BINARY_DIR}/caffe/libiomp5.so"
            "${CMAKE_BINARY_DIR}/caffe/libmklml_intel.so"
            "${CMAKE_BINARY_DIR}/caffe/lib/libmkldnn.so"
            )
685
        else (USE_MKL)
G
gineshidalgo99 已提交
686
          execute_process(COMMAND git checkout master WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
687
        endif (USE_MKL)
G
gineshidalgo99 已提交
688 689
        set(CAFFE_CPU_ONLY ON)
        set(USE_CUDNN OFF)
690 691 692
      elseif (${GPU_MODE} MATCHES "OPENCL")
        execute_process(COMMAND git checkout opencl WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
        set(USE_CUDNN OFF)
693
      endif (${GPU_MODE} MATCHES "CUDA")
G
gineshidalgo99 已提交
694

695 696
      # Build Caffe
      message(STATUS "Caffe will be built from source now.")
B
Bikramjot Hanzra 已提交
697
      find_package(Caffe)
698 699 700
      include(ExternalProject)
      set(CAFFE_PREFIX caffe)
      set(CAFFE_URL ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
B
Bikramjot Hanzra 已提交
701

G
gineshidalgo99 已提交
702 703
      # One for Intel Branch and one for Master
      if (USE_MKL)
704
        ExternalProject_Add(openpose_lib
G
gineshidalgo99 已提交
705 706 707 708
          SOURCE_DIR ${CAFFE_URL}
          PREFIX ${CAFFE_PREFIX}
          CMAKE_ARGS -DCMAKE_INSTALL_PREFIX:PATH=<INSTALL_DIR>
          -DMKLDNN_INSTALL_DIR:PATH=<INSTALL_DIR>
709
          -DUSE_MKL2017_AS_DEFAULT_ENGINE=${CAFFE_CPU_ONLY}
G
gineshidalgo99 已提交
710
          -DUSE_CUDNN=${USE_CUDNN}
711
          -DCUDA_ARCH_NAME=${CUDA_ARCH}
712 713
          -DCUDA_ARCH_BIN=${CUDA_ARCH_BIN}
          -DCUDA_ARCH_PTX=${CUDA_ARCH_PTX}
G
gineshidalgo99 已提交
714 715 716
          -DCPU_ONLY=${CAFFE_CPU_ONLY}
          -DCMAKE_BUILD_TYPE=Release
          -DBUILD_docs=OFF
R
Raaj 已提交
717
          -DBUILD_python=OFF
G
gineshidalgo99 已提交
718 719 720 721 722
          -DBUILD_python_layer=OFF
          -DUSE_LEVELDB=OFF
          -DUSE_LMDB=OFF
          -DUSE_OPENCV=OFF)
          # -DOpenCV_DIR=${OpenCV_DIR})
723
      else (USE_MKL)
724
        ExternalProject_Add(openpose_lib
725 726
          SOURCE_DIR ${CAFFE_URL}
          PREFIX ${CAFFE_PREFIX}
G
gineshidalgo99 已提交
727
          CMAKE_ARGS -DCMAKE_INSTALL_PREFIX:PATH=<INSTALL_DIR>
G
gineshidalgo99 已提交
728
          -DUSE_CUDNN=${USE_CUDNN}
729
          -DCUDA_ARCH_NAME=${CUDA_ARCH}
730 731
          -DCUDA_ARCH_BIN=${CUDA_ARCH_BIN}
          -DCUDA_ARCH_PTX=${CUDA_ARCH_PTX}
G
gineshidalgo99 已提交
732 733 734
          -DCPU_ONLY=${CAFFE_CPU_ONLY}
          -DCMAKE_BUILD_TYPE=Release
          -DBUILD_docs=OFF
R
Raaj 已提交
735
          -DBUILD_python=OFF
G
gineshidalgo99 已提交
736 737 738 739 740
          -DBUILD_python_layer=OFF
          -DUSE_LEVELDB=OFF
          -DUSE_LMDB=OFF
          -DUSE_OPENCV=OFF)
          # -DOpenCV_DIR=${OpenCV_DIR})
741
      endif (USE_MKL)
B
Bikramjot Hanzra 已提交
742

743
      ExternalProject_Get_Property(openpose_lib install_dir)
744

B
Bikramjot Hanzra 已提交
745
      if (NOT Caffe_FOUND)
746
        add_custom_command(TARGET openpose_lib
B
Bikramjot Hanzra 已提交
747
            POST_BUILD
748
            COMMAND ${CMAKE_COMMAND} ${CMAKE_SOURCE_DIR}
B
Bikramjot Hanzra 已提交
749 750 751 752
            COMMAND $(MAKE)
            WORKING_DIRECTORY ${CMAKE_BINARY_DIR}
            COMMENT "Rerunning cmake after building Caffe submodule")
      endif (NOT Caffe_FOUND)
753 754 755 756

    endif (BUILD_CAFFE)

    if (NOT Caffe_FOUND AND NOT BUILD_CAFFE)
G
gineshidalgo99 已提交
757
      message(FATAL_ERROR "Caffe not found. Either turn on the BUILD_CAFFE option or specify the path of Caffe includes
758
        and libs using -DCaffe_INCLUDE_DIRS and -DCaffe_LIBS.")
759 760 761 762
    endif (NOT Caffe_FOUND AND NOT BUILD_CAFFE)

  endif (${DL_FRAMEWORK} MATCHES "CAFFE")

763
endif (UNIX OR APPLE)
B
Bikramjot Hanzra 已提交
764 765 766 767

### PROJECT INCLUDES
# Specify the include directories
include_directories(
768
  include
J
Jim Carroll 已提交
769
  ${Protobuf_INCLUDE_DIRS}
B
Bikramjot Hanzra 已提交
770 771 772
  ${GFLAGS_INCLUDE_DIR}
  ${GLOG_INCLUDE_DIR}
  ${OpenCV_INCLUDE_DIRS})
B
Bikramjot Hanzra 已提交
773

774 775 776 777 778 779 780 781
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)

782 783 784 785 786 787
# Calibration
if (NOT ${WITH_EIGEN} MATCHES "NONE")
  include_directories(
      ${EIGEN3_INCLUDE_DIRS})
endif (NOT ${WITH_EIGEN} MATCHES "NONE")

788 789 790 791 792
if (APPLE)
  include_directories(
      "/usr/local/opt/openblas/include")
endif (APPLE)

793
if (USE_MKL)
794 795 796 797
 include_directories(
     "${MKL_PATH}/include/")
endif (USE_MKL)

B
Bikramjot Hanzra 已提交
798 799 800 801 802
if (Caffe_FOUND)
  include_directories(
      ${Caffe_INCLUDE_DIRS})
endif (Caffe_FOUND)

G
gineshidalgo99 已提交
803 804
if (${GPU_MODE} MATCHES "CUDA")
  include_directories(
B
Bikramjot Hanzra 已提交
805
      ${CUDA_INCLUDE_DIRS})
806
elseif (${GPU_MODE} MATCHES "OPENCL")
807 808
  include_directories(
    ${OpenCL_INCLUDE_DIRS})
809
endif (${GPU_MODE} MATCHES "CUDA")
810
# 3D
811 812 813
if (WITH_3D_RENDERER)
  include_directories(${GLUT_INCLUDE_DIRS})
endif (WITH_3D_RENDERER)
814 815 816
if (WITH_CERES)
  include_directories(${CERES_INCLUDE_DIRS})
endif (WITH_CERES)
817
if (WITH_FLIR_CAMERA)
G
gineshidalgo99 已提交
818
  include_directories(SYSTEM ${SPINNAKER_INCLUDE_DIRS}) # To remove its warnings, equiv. to -isystem
819
endif (WITH_FLIR_CAMERA)
820
if (WITH_3D_ADAM_MODEL)
G
gineshidalgo99 已提交
821
                                    include_directories(include/adam) # TODO: TEMPORARY - TO BE REMOVED IN THE FUTURE
822 823 824 825
  include_directories(${CERES_INCLUDE_DIRS})
  include_directories(${EIGEN3_INCLUDE_DIRS})
  include_directories(${IGL_INCLUDE_DIRS})
  include_directories(${LIBIGL_INCLUDE_DIRS})
826
  include_directories(${GLUT_INCLUDE_DIRS} ${GLEW_INCLUDE_DIRS} ${OPENGL_INCLUDE_DIR})
827
endif (WITH_3D_ADAM_MODEL)
828
# Windows includes
829 830 831 832 833
if (WIN32)
  include_directories(
    ${Boost_INCLUDE_DIRS}
    ${WINDOWS_INCLUDE_DIRS})
endif (WIN32)
B
Bikramjot Hanzra 已提交
834 835


836 837
### COLLECT ALL 3RD-PARTY LIBRARIES TO BE LINKED AGAINST
set(OpenPose_3rdparty_libraries ${OpenCV_LIBS} ${GLOG_LIBRARY})
838
if (UNIX OR APPLE)
839 840 841 842 843
  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})
844
endif (UNIX OR APPLE)
845 846 847 848 849 850 851 852 853 854 855
# 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 已提交
856 857 858 859 860 861 862 863 864
# 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)
865
# 3-D
866 867 868 869
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)
870
if (WITH_3D_RENDERER)
871
  set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries} ${GLUT_LIBRARY} ${OPENGL_LIBRARIES})
872 873 874 875 876 877 878
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 已提交
879
# Pthread
880
if (UNIX OR APPLE)
G
gineshidalgo99 已提交
881
  set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries} pthread)
882
endif (UNIX OR APPLE)
883 884 885 886

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


B
Bikramjot Hanzra 已提交
887 888 889 890 891 892 893
### ADD SUBDIRECTORIES
if (Caffe_FOUND)
  add_subdirectory(src)
  if (BUILD_EXAMPLES)
    add_subdirectory(examples)
  endif (BUILD_EXAMPLES)
endif (Caffe_FOUND)
B
Bikramjot Hanzra 已提交
894 895


G
gineshidalgo99 已提交
896
### DOWNLOAD MODELS
B
Bikramjot Hanzra 已提交
897 898 899 900 901
# 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 已提交
902

903 904 905
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 已提交
906
  5156d31f670511fce9b4e28b403f2939) # Body (COCO)
907
download_model("body (MPI)" ${DOWNLOAD_BODY_MPI_MODEL} pose/mpi/pose_iter_160000.caffemodel
G
gineshidalgo99 已提交
908
  2ca0990c7562bd7ae03f3f54afa96e00) # Body (MPI)
909
download_model("face" ${DOWNLOAD_FACE_MODEL} face/pose_iter_116000.caffemodel
G
gineshidalgo99 已提交
910
  e747180d728fa4e4418c465828384333) # Face
911
download_model("hand" ${DOWNLOAD_HAND_MODEL} hand/pose_iter_102000.caffemodel
G
gineshidalgo99 已提交
912
  a82cfc3fea7c62f159e11bd3674c1531) # Hand
B
Bikramjot Hanzra 已提交
913 914

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

R
Raaj 已提交
916

R
Raaj 已提交
917
### PYTHON
R
Raaj 已提交
918 919 920 921
if (BUILD_PYTHON)
  if (WIN32)
    execute_process(COMMAND cmd /c cd ${CMAKE_SOURCE_DIR} & git submodule update --init 3rdparty/pybind11/)
    add_subdirectory(3rdparty/pybind11)
R
Raaj 已提交
922
    add_subdirectory(python)
R
Raaj 已提交
923 924 925
  elseif (UNIX OR APPLE)
    if (Caffe_FOUND)
      execute_process(COMMAND git submodule update --init ${CMAKE_SOURCE_DIR}/3rdparty/pybind11/)
926
      # execute_process(COMMAND git submodule update --init --recursive --remote) # This would initialize them all
R
Raaj 已提交
927 928 929 930 931 932 933 934
      add_subdirectory(3rdparty/pybind11)
      add_subdirectory(python)
    endif (Caffe_FOUND)
  else (WIN32)
    message(FATAL_ERROR "Unknown OS.")
  endif (WIN32)
endif (BUILD_PYTHON)

935

G
gineshidalgo99 已提交
936
### GENERATE DOCUMENTATION
937
if (UNIX OR APPLE)
938 939 940 941 942 943 944 945 946

  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 已提交
947 948 949 950
        COMMAND ${DOXYGEN_EXECUTABLE} ${DOXYGEN_FILE}
        WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/doc
        COMMENT "Generating API documentation with Doxygen"
        VERBATIM)
951
    else (DOXYGEN_FOUND)
952
      message(FATAL_ERROR "Doxygen needs to be installed to generate the doxygen documentation.")
953 954 955
    endif (DOXYGEN_FOUND)
  endif (BUILD_DOCS)

956
endif (UNIX OR APPLE)
957

B
bikz05 已提交
958

G
gineshidalgo99 已提交
959
### INSTALL
960
if (UNIX OR APPLE)
B
Bikramjot Hanzra 已提交
961 962 963 964 965 966 967 968 969 970 971 972 973 974 975 976 977 978 979 980 981 982 983 984 985 986
  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)
987
endif (UNIX OR APPLE)