CMakeLists.txt 38.2 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
G
Gines Hidalgo 已提交
13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33
# Ubuntu 18 default. After 3.8, no need for find_CUDA
# https://cmake.org/cmake/help/v3.10/module/FindCUDA.html
# https://cmake.org/cmake/help/v3.10/command/project.html
# https://devblogs.nvidia.com/building-cuda-applications-cmake/
if (${CMAKE_VERSION} VERSION_GREATER 3.9.0)
  cmake_policy(SET CMP0048 NEW)
  project(OpenPose VERSION ${OpenPose_VERSION})
  # # Not tested
  # cmake_policy(SET CMP0048 NEW)
  # set(CUDACXX /usr/local/cuda/bin/nvcc)
  # project(OpenPose VERSION ${OpenPose_VERSION} LANGUAGES CXX CUDA)
  # set(AUTO_FOUND_CUDA TRUE)
  # # else
  # set(AUTO_FOUND_CUDA FALSE)
# Ubuntu 16 default
elseif (${CMAKE_VERSION} VERSION_GREATER 3.0.0)
  cmake_policy(SET CMP0048 NEW)
  project(OpenPose VERSION ${OpenPose_VERSION})
else (${CMAKE_VERSION} VERSION_GREATER 3.9.0)
  project(OpenPose)
endif (${CMAKE_VERSION} VERSION_GREATER 3.9.0)
34 35 36
cmake_minimum_required(VERSION 2.8.7 FATAL_ERROR) # min. cmake version recommended by Caffe


37 38
### Extra functionality
include(cmake/Utils.cmake)
G
gineshidalgo99 已提交
39 40 41
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 已提交
42 43


G
gineshidalgo99 已提交
44
### CMAKE_BUILD_TYPE
45
# Default: Release
G
gineshidalgo99 已提交
46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65
# 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")
66 67


G
gineshidalgo99 已提交
68
### FLAGS
69
if (WIN32)
G
gineshidalgo99 已提交
70
  # TODO -- Makeshift solution -- This prevents rerunning build again
71 72 73
  # https://gitlab.kitware.com/cmake/cmake/issues/16783
  set(CMAKE_SUPPRESS_REGENERATION ON)

A
arkceajin 已提交
74 75
  string (REPLACE "/D_WINDOWS" "" CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}")
  string (REPLACE "/DWIN32" "" CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}")
76 77 78 79

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

G
gineshidalgo99 已提交
80
  set(CMAKE_CONFIGURATION_TYPES Release Debug CACHE TYPE INTERNAL FORCE)
81

82 83 84 85 86 87 88 89 90
  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")
91

92 93 94
  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}")
95
elseif (UNIX)
G
gineshidalgo99 已提交
96
  # Turn on C++11
97 98
  add_definitions(-std=c++11)
  set(CMAKE_CXX_FLAGS_RELEASE "-O3")
99 100 101 102
elseif (APPLE)
  # Turn on C++11
  set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11")
  set(CMAKE_CXX_FLAGS_RELEASE "-O3")
103
endif (WIN32)
B
Bikramjot Hanzra 已提交
104 105

# C++ additional flags
G
gineshidalgo99 已提交
106
if (CMAKE_COMPILER_IS_GNUCXX)
107 108 109 110 111 112 113 114

  # 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")
115
  endif (${INSTRUCTION_SET} MATCHES "SSE4")
116 117 118 119

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

G
gineshidalgo99 已提交
122
  message(STATUS "GCC detected, adding compile flags")
123
  set(OP_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${SIMD_FLAGS} -fopenmp -Wpedantic -Wall -Wextra -Wfatal-errors")
G
gineshidalgo99 已提交
124
endif (CMAKE_COMPILER_IS_GNUCXX)
B
Bikramjot Hanzra 已提交
125 126


G
gineshidalgo99 已提交
127
### PROJECT OPTIONS
B
Bikramjot Hanzra 已提交
128 129 130 131 132 133 134 135 136
# 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 已提交
137
    "DL_FRAMEWORK" ON)
138 139
  # OpenPose flags
  add_definitions(-DUSE_CAFFE)
B
Bikramjot Hanzra 已提交
140 141
endif (${DL_FRAMEWORK} MATCHES "CAFFE")

142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160
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 已提交
161
# Set the acceleration library
G
gineshidalgo99 已提交
162 163 164 165 166
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)
167 168 169 170
# 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 已提交
171
#   set_property(CACHE GPU_MODE PROPERTY STRINGS CUDA CPU_ONLY)
172 173 174 175
# elseif (OpenCL_FOUND)
#   set_property(CACHE GPU_MODE PROPERTY STRINGS OPENCL CPU_ONLY)
# else ()
#   set_property(CACHE GPU_MODE PROPERTY STRINGS CPU_ONLY)
176
# endif (CUDA_FOUND AND OpenCL_FOUND)
R
Raaj 已提交
177
set_property(CACHE GPU_MODE PROPERTY STRINGS CUDA OPENCL CPU_ONLY)
G
gineshidalgo99 已提交
178

179
# Look for CUDA
G
gineshidalgo99 已提交
180
set(CUDA_FOUND FALSE)
181 182 183 184 185
if (${GPU_MODE} MATCHES "CUDA")
  find_package(CUDA)
endif (${GPU_MODE} MATCHES "CUDA")
# Look for OpenCL
set(OpenCL_FOUND FALSE)
G
gineshidalgo99 已提交
186
set(CUDA_VERSION_MAJOR 0)
187 188 189 190
if (${GPU_MODE} MATCHES "OPENCL")
  find_package(OpenCL)
endif (${GPU_MODE} MATCHES "OPENCL")

191 192 193 194 195 196 197
# 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)
198
endif (CUDA_FOUND AND OpenCL_FOUND AND ${CUDA_VERSION_MAJOR} LESS 9)
199
# Handle desired GPU mode option
200 201 202
if (${GPU_MODE} MATCHES "CUDA")
  # OpenPose flags
  add_definitions(-DUSE_CUDA)
G
gineshidalgo99 已提交
203
  message(STATUS "Building with CUDA.")
204
elseif (${GPU_MODE} MATCHES "CPU_ONLY")
G
gineshidalgo99 已提交
205 206
  add_definitions(-DUSE_CPU_ONLY)
  message(STATUS "Building CPU Only.")
207 208
  # OpenPose flag for Caffe
  add_definitions(-DCPU_ONLY)
209 210 211 212 213
elseif (${GPU_MODE} MATCHES "OPENCL")
  # OpenPose flag for Caffe
  add_definitions(-DUSE_OPENCL)
  add_definitions(-DUSE_GREENTEA)
  message(STATUS "Building with OpenCL.")
214
endif (${GPU_MODE} MATCHES "CUDA")
G
gineshidalgo99 已提交
215 216 217 218 219

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

223 224 225 226 227 228
if (${USE_MKL})
  # OpenPose flags
  add_definitions(-DUSE_MKL)
  message(STATUS "Building with MKL support.")
endif (${USE_MKL})

229 230 231 232 233
# Set/disable profiler
if (PROFILER_ENABLED)
  add_definitions(-DPROFILER_ENABLED)
endif (PROFILER_ENABLED)

B
Bikramjot Hanzra 已提交
234 235 236 237 238 239 240 241 242 243
# 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")
244

G
gineshidalgo99 已提交
245
# Suboptions for OpenPose 3D Reconstruction module and demo
246
option(WITH_3D_RENDERER "Add OpenPose 3D renderer module (it requires FreeGLUT library)." OFF)
247
if (UNIX AND NOT APPLE)
G
gineshidalgo99 已提交
248
  option(WITH_CERES "Add Ceres support for advanced 3-D reconstruction." OFF)
249
endif (UNIX AND NOT APPLE)
250
option(WITH_FLIR_CAMERA "Add FLIR (formerly Point Grey) camera code (requires Spinnaker SDK already installed)." OFF)
G
gineshidalgo99 已提交
251
# option(WITH_3D_ADAM_MODEL "Add 3-D Adam model (requires OpenGL, Ceres, Eigen, OpenMP, FreeImage, GLEW, and IGL already installed)." OFF)
252

G
gineshidalgo99 已提交
253
# Faster GUI rendering
254 255 256
# 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 已提交
257

258
# Set the acceleration library
G
gineshidalgo99 已提交
259
if (WIN32 OR APPLE)
260 261 262 263 264
  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 已提交
265
endif (WIN32 OR APPLE)
266

G
gineshidalgo99 已提交
267 268 269 270 271
# # 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 已提交
272
# Download the models
273 274 275
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 已提交
276
option(DOWNLOAD_FACE_MODEL "Download face model." ON)
277 278 279
option(DOWNLOAD_HAND_MODEL "Download hand model." ON)

# Asio
G
gineshidalgo99 已提交
280
# option(USE_ASIO "Include Asio header-only library." OFF)
B
Bikramjot Hanzra 已提交
281 282 283 284

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

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

B
bikz05 已提交
293
# Build as shared library
R
Raaj 已提交
294
option(BUILD_SHARED_LIBS "Build as shared lib." ON)
B
bikz05 已提交
295

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

G
gineshidalgo99 已提交
299
# Threads - Pthread
300
if (${GPU_MODE} MATCHES "OPENCL" OR (UNIX OR APPLE))
301 302
  unset(CMAKE_THREAD_LIBS_INIT CACHE)
  find_package(Threads)
303 304 305 306
endif (${GPU_MODE} MATCHES "OPENCL" OR (UNIX OR APPLE))

# Caffe OpenCL Boost Issue
if (APPLE)
R
Raaj 已提交
307
  if (${GPU_MODE} MATCHES "OPENCL" OR BUILD_PYTHON)
308 309 310 311 312
    unset(Boost_SYSTEM_LIBRARY CACHE)
    find_package(Boost COMPONENTS system REQUIRED)
  else ()
    set(Boost_SYSTEM_LIBRARY "")
  endif ()
G
gineshidalgo99 已提交
313
endif (APPLE)
B
Bikramjot Hanzra 已提交
314

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

G
gineshidalgo99 已提交
318 319
if (${GPU_MODE} MATCHES "CUDA")
  find_package(CUDA)
320
endif (${GPU_MODE} MATCHES "CUDA")
G
gineshidalgo99 已提交
321

322
# Adding 3D
G
gineshidalgo99 已提交
323 324 325 326
if (WITH_OPENCV_WITH_OPENGL)
  # OpenPose flags
  add_definitions(-DUSE_OPENCV_WITH_OPENGL)
endif (WITH_OPENCV_WITH_OPENGL)
327
if (WITH_3D_RENDERER)
328
  # OpenPose flags
329
  add_definitions(-DUSE_3D_RENDERER)
330
endif (WITH_3D_RENDERER)
331
if (WITH_CERES)
332
  add_definitions(-DUSE_CERES)
333
endif (WITH_CERES)
334 335
if (WITH_FLIR_CAMERA)
  # OpenPose flags
336
  add_definitions(-DUSE_FLIR_CAMERA)
337
endif (WITH_FLIR_CAMERA)
338 339 340 341
if (WITH_3D_ADAM_MODEL)
  # OpenPose flags
  add_definitions(-DUSE_3D_ADAM_MODEL)
endif (WITH_3D_ADAM_MODEL)
342

G
gineshidalgo99 已提交
343 344 345
# Adding tracking
if (WITH_TRACKING)
  # OpenPose flags
346
  add_definitions(-DUSE_TRACKING)
G
gineshidalgo99 已提交
347 348
endif (WITH_TRACKING)

G
gineshidalgo99 已提交
349 350 351 352 353 354
# Unity
if (BUILD_UNITY_SUPPORT)
  # OpenPose flags
  add_definitions(-DUSE_UNITY_SUPPORT)
endif (BUILD_UNITY_SUPPORT)

355 356 357 358 359 360 361 362 363
# 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
364
  add_definitions(-DUSE_EIGEN)
365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380
  # 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")

381
if (UNIX OR APPLE)
G
gineshidalgo99 已提交
382 383 384
  if (${GPU_MODE} MATCHES "CUDA")
    include(cmake/Cuda.cmake)
    find_package(CuDNN)
385
  endif (${GPU_MODE} MATCHES "CUDA")
J
Jim Carroll 已提交
386 387 388 389
  find_package(GFlags) # For Caffe and OpenPose
  find_package(Glog) # For Caffe
  find_package(Protobuf REQUIRED) # For Caffe

B
Bikramjot Hanzra 已提交
390 391
  if (OpenCV_CONFIG_FILE)
    include (${OpenCV_CONFIG_FILE})
J
Jim Carroll 已提交
392 393 394
  # Allow explicitly setting the OpenCV includes and libs
  elseif (OpenCV_INCLUDE_DIRS AND OpenCV_LIBS)
    set(OpenCV_FOUND 1)
B
Bikramjot Hanzra 已提交
395 396 397
  elseif (OpenCV_INCLUDE_DIRS AND OpenCV_LIBS_DIR)
    file(GLOB_RECURSE OpenCV_LIBS "${OpenCV_LIBS_DIR}*.so")
    set(OpenCV_FOUND 1)
398
  else (OpenCV_CONFIG_FILE)
B
Bikramjot Hanzra 已提交
399 400
    find_package(OpenCV)
  endif (OpenCV_CONFIG_FILE)
401

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

G
gineshidalgo99 已提交
445 446 447 448
  # OpenMP
  if (${GPU_MODE} MATCHES "CPU_ONLY")
    find_package(OpenMP)
    if (OPENMP_FOUND)
G
gineshidalgo99 已提交
449 450 451
      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}")
452 453
    endif (OPENMP_FOUND)
  endif (${GPU_MODE} MATCHES "CPU_ONLY")
G
gineshidalgo99 已提交
454 455 456 457 458 459 460

  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.")
461
      execute_process(COMMAND cat install_cuda.sh WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/scripts/ubuntu)
G
gineshidalgo99 已提交
462 463 464 465 466
      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.")
467
      execute_process(COMMAND cat install_cudnn.sh WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/scripts/ubuntu)
G
gineshidalgo99 已提交
468 469
      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)
470
  endif (${GPU_MODE} MATCHES "CUDA")
471 472

  if (NOT GLOG_FOUND)
G
gineshidalgo99 已提交
473 474
    message(FATAL_ERROR "Glog not found. Install Glog from the command line using the command(s) -\
      sudo apt-get install libgoogle-glog-dev")
475 476 477
  endif (NOT GLOG_FOUND)

  if (NOT GFLAGS_FOUND)
G
gineshidalgo99 已提交
478
    message(FATAL_ERROR "GFlags not found. Install GFlags from the command line using the command(s) --\
479 480 481 482 483 484 485 486
      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)

487
endif (UNIX OR APPLE)
488 489 490

if (WIN32)

G
gineshidalgo99 已提交
491
  if ("${CMAKE_VERSION}" VERSION_GREATER 3.6.3)
492
    set_property(DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR} PROPERTY VS_STARTUP_PROJECT OpenPoseDemo)
G
gineshidalgo99 已提交
493
  endif ("${CMAKE_VERSION}" VERSION_GREATER 3.6.3)
494 495 496 497 498 499 500 501 502

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

503
  # Download required zip files
504 505 506
  download_zip("opencv_401_v14_15_2019_03_14.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} df8f611c032cfa86c56a8ad419f01832)
  # download_zip("caffe3rdparty_2019_03_12.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} 34c49be6a2418fd1670b46fc455eebc0)
  download_zip("caffe3rdparty_15_2019_03_14.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} 9fcc6bdf4765db699d1a279c8d6b830b)
507 508
  if (${GPU_MODE} MATCHES "OPENCL")
    download_zip("caffe_opencl_2018_02_13.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} 3ac3e1acf5ee6a4e57920be73053067a)
509 510
  elseif (${GPU_MODE} MATCHES "CPU_ONLY")
    download_zip("caffe_cpu_2018_05_27.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} 87E8401B6DFBAC5B8E909DD20E3B3390)
511
  else (${GPU_MODE} MATCHES "OPENCL")
512 513
    # download_zip("caffe_2019_03_12.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} 859a592310f0928fd4f40da1456a217f)
    download_zip("caffe_15_2019_03_14.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} f2d8beee9e4959fa419dcb54d507107c)
514
  endif (${GPU_MODE} MATCHES "OPENCL")
515
  if (WITH_3D_RENDERER)
516
    download_zip("freeglut_2018_01_14.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} BB182187285E06880F0EDE3A39530091)
517
  endif (WITH_3D_RENDERER)
518 519
  message(STATUS "Windows dependencies downloaded.")

520 521 522
  # set(VS_VERSION "14")
  set(VS_VERSION "15") # Note: VS2017 code seems to simply works in VS2015
  find_library(OpenCV_LIBS opencv_world401 HINTS ${FIND_LIB_PREFIX}/opencv/x64/vc${VS_VERSION}/lib)
523 524 525 526
  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)
527

528 529 530
  # If OpenPose builds it
  if (BUILD_CAFFE)
    unset(Caffe_INCLUDE_DIRS CACHE)
531 532
    unset(Caffe_LIB CACHE)
    unset(Caffe_Proto_LIB CACHE)
533 534 535 536
  endif (BUILD_CAFFE)
  # OpenCL
  if (${GPU_MODE} MATCHES "OPENCL")
    set(VCXPROJ_FILE_GPU_MODE "_CL")
537 538
    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)
539
  # CPU & CUDA
540
  else (${GPU_MODE} MATCHES "OPENCL")
541 542 543 544 545 546 547 548 549 550 551
    # 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")
552
  endif (${GPU_MODE} MATCHES "OPENCL")
R
Raaj 已提交
553 554
  # Boost DepCopy over required DLL F
  if (${GPU_MODE} MATCHES "CPU_ONLY" OR ${GPU_MODE} MATCHES "OPENCL" OR BUILD_PYTHON)
555 556 557 558
      find_library(BOOST_SYSTEM_LIB_RELEASE libboost_system-vc141-mt-x64-1_69 HINTS ${FIND_LIB_PREFIX}/caffe3rdparty/lib)
      find_library(BOOST_SYSTEM_LIB_DEBUG libboost_system-vc141-mt-gd-x64-1_69 HINTS ${FIND_LIB_PREFIX}/caffe3rdparty/lib)
      find_library(BOOST_FILESYSTEM_LIB_RELEASE libboost_filesystem-vc141-mt-x64-1_69 HINTS ${FIND_LIB_PREFIX}/caffe3rdparty/lib)
      find_library(BOOST_FILESYSTEM_LIB_DEBUG libboost_filesystem-vc141-mt-gd-x64-1_69 HINTS ${FIND_LIB_PREFIX}/caffe3rdparty/lib)
R
Raaj 已提交
559 560 561 562 563 564
  else ()
      set(BOOST_SYSTEM_LIB_RELEASE "")
      set(BOOST_SYSTEM_LIB_DEBUG "")
      set(BOOST_FILESYSTEM_LIB_RELEASE "")
      set(BOOST_FILESYSTEM_LIB_DEBUG "")
  endif ()
565
  if (WITH_3D_RENDERER)
566 567
    find_library(GLUT_LIBRARY freeglut HINTS ${FIND_LIB_PREFIX}/freeglut/lib)
    message(STATUS "\${GLUT_LIBRARY} = ${GLUT_LIBRARY}")
568 569 570 571
  endif (WITH_3D_RENDERER)
  if (WITH_FLIR_CAMERA)
    find_library(SPINNAKER_LIB spinnaker_v140 HINTS ${FIND_LIB_PREFIX}/spinnaker/lib)
  endif (WITH_FLIR_CAMERA)
572 573
  set(Caffe_LIBS ${Caffe_LIB};${Caffe_Proto_LIB})
  set(OpenCV_INCLUDE_DIRS "3rdparty/windows/opencv/include")
574
  # OpenCL
575 576 577
  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")
578
  # CUDA and CPU
579
  else (${GPU_MODE} MATCHES "OPENCL")
580 581 582 583 584 585 586
    # 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")
587
  endif (${GPU_MODE} MATCHES "OPENCL")
588
  # set(Boost_INCLUDE_DIRS "3rdparty/windows/caffe3rdparty/include/boost-1_61") # Added in WINDOWS_INCLUDE_DIRS
589
  set(WINDOWS_INCLUDE_DIRS "3rdparty/windows/caffe3rdparty/include")
590
  if (WITH_3D_RENDERER)
591
    set(GLUT_INCLUDE_DIRS "3rdparty/windows/freeglut/include")
592 593 594 595
  endif (WITH_3D_RENDERER)
  if (WITH_FLIR_CAMERA)
    set(SPINNAKER_INCLUDE_DIRS "3rdparty/windows/spinnaker/include")
  endif (WITH_FLIR_CAMERA)
B
Bikramjot Hanzra 已提交
596
  set(Caffe_FOUND 1)
R
Raaj 已提交
597 598 599

  # Build DLL Must be on if Build Python is on
  if (BUILD_PYTHON)
G
gineshidalgo99 已提交
600 601
    if (NOT BUILD_BIN_FOLDER)
      message(FATAL_ERROR "BUILD_BIN_FOLDER must be turned on to as well to build python library")
G
gineshidalgo99 已提交
602 603
    endif (NOT BUILD_BIN_FOLDER)
  endif (BUILD_PYTHON)
R
Raaj 已提交
604 605

  # Auto copy DLLs
G
gineshidalgo99 已提交
606 607 608
  if (BUILD_BIN_FOLDER)
    # Locate DLLs
    # Caffe DLLs
R
Raaj 已提交
609 610 611 612 613 614 615
    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 已提交
616
    # Caffe 3rd-party DLLs
617
    file(GLOB CAFFE_3RD_PARTY_DLL "${CMAKE_SOURCE_DIR}/3rdparty/windows/caffe3rdparty/lib/*.dll")
G
gineshidalgo99 已提交
618
    # OpenCV DLLs
619
    file(GLOB OPENCV_DLL "${CMAKE_SOURCE_DIR}/3rdparty/windows/opencv/x64/vc${VS_VERSION}/bin/*.dll")
620 621 622 623
    # 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 已提交
624 625 626 627 628
    # 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})
629 630 631
    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 已提交
632
  endif (BUILD_BIN_FOLDER)
R
Raaj 已提交
633

634
endif (WIN32)
B
Bikramjot Hanzra 已提交
635 636


G
gineshidalgo99 已提交
637
### CAFFE
638
if (UNIX OR APPLE)
B
Bikramjot Hanzra 已提交
639

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

642 643 644 645 646 647
    # 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 已提交
648

649 650
    # Else build from scratch
    if (BUILD_CAFFE)
B
Bikramjot Hanzra 已提交
651

652 653
      # Download Caffe
      message(STATUS "Caffe will be downloaded from source now. NOTE: This process might take several minutes depending
G
gineshidalgo99 已提交
654 655 656 657 658 659
        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 已提交
660
        execute_process(COMMAND git submodule update --init ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
661
        # execute_process(COMMAND git submodule update --init --recursive --remote) # This would initialize them all
662
      else (CAFFE_DIR_VALID_LENGTH EQUAL 0)
G
gineshidalgo99 已提交
663
        message(STATUS "Caffe has already been downloaded.")
664
      endif (CAFFE_DIR_VALID_LENGTH EQUAL 0)
G
gineshidalgo99 已提交
665 666 667 668

      # Build Process
      set(CAFFE_CPU_ONLY OFF)
      if (${GPU_MODE} MATCHES "CUDA")
G
gineshidalgo99 已提交
669 670
        # execute_process(COMMAND git checkout master WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
        execute_process(COMMAND git checkout b5ede48 WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
G
gineshidalgo99 已提交
671 672
      elseif (${GPU_MODE} MATCHES "CPU_ONLY")
        if (USE_MKL)
G
gineshidalgo99 已提交
673 674
          #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 已提交
675 676
          execute_process(COMMAND sh prepare_mkl.sh WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/3rdparty/caffe/external/mkl
              OUTPUT_VARIABLE rv)
G
gineshidalgo99 已提交
677 678 679
          set( MLIST ${rv} )
          separate_arguments(MLIST)
          list(GET MLIST 0 MKL_PATH)
680
          message(STATUS ${MKL_PATH})
G
gineshidalgo99 已提交
681 682 683 684 685
          file(GLOB MKL_SO
            "${MKL_PATH}lib/*"
            )
          file(COPY ${MKL_SO} DESTINATION ${CMAKE_BINARY_DIR}/caffe)

G
gineshidalgo99 已提交
686 687 688 689 690 691 692 693 694 695 696 697
          # 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 已提交
698
          set(MKL_LIBS
G
gineshidalgo99 已提交
699
            #"${CMAKE_BINARY_DIR}/caffe/libmlsl.so"
G
gineshidalgo99 已提交
700 701 702 703
            "${CMAKE_BINARY_DIR}/caffe/libiomp5.so"
            "${CMAKE_BINARY_DIR}/caffe/libmklml_intel.so"
            "${CMAKE_BINARY_DIR}/caffe/lib/libmkldnn.so"
            )
704
        else (USE_MKL)
G
gineshidalgo99 已提交
705 706
          # execute_process(COMMAND git checkout master WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
          execute_process(COMMAND git checkout b5ede48 WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
707
        endif (USE_MKL)
G
gineshidalgo99 已提交
708 709
        set(CAFFE_CPU_ONLY ON)
        set(USE_CUDNN OFF)
710
      elseif (${GPU_MODE} MATCHES "OPENCL")
R
Raaj 已提交
711
        execute_process(COMMAND git checkout fe2a1102 WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
712
        set(USE_CUDNN OFF)
713
      endif (${GPU_MODE} MATCHES "CUDA")
G
gineshidalgo99 已提交
714

715 716
      # Build Caffe
      message(STATUS "Caffe will be built from source now.")
B
Bikramjot Hanzra 已提交
717
      find_package(Caffe)
718 719 720
      include(ExternalProject)
      set(CAFFE_PREFIX caffe)
      set(CAFFE_URL ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
B
Bikramjot Hanzra 已提交
721

G
gineshidalgo99 已提交
722 723
      # One for Intel Branch and one for Master
      if (USE_MKL)
724
        ExternalProject_Add(openpose_lib
G
gineshidalgo99 已提交
725 726 727 728
          SOURCE_DIR ${CAFFE_URL}
          PREFIX ${CAFFE_PREFIX}
          CMAKE_ARGS -DCMAKE_INSTALL_PREFIX:PATH=<INSTALL_DIR>
          -DMKLDNN_INSTALL_DIR:PATH=<INSTALL_DIR>
729
          -DUSE_MKL2017_AS_DEFAULT_ENGINE=${CAFFE_CPU_ONLY}
G
gineshidalgo99 已提交
730
          -DUSE_CUDNN=${USE_CUDNN}
731
          -DCUDA_ARCH_NAME=${CUDA_ARCH}
732 733
          -DCUDA_ARCH_BIN=${CUDA_ARCH_BIN}
          -DCUDA_ARCH_PTX=${CUDA_ARCH_PTX}
G
gineshidalgo99 已提交
734 735 736
          -DCPU_ONLY=${CAFFE_CPU_ONLY}
          -DCMAKE_BUILD_TYPE=Release
          -DBUILD_docs=OFF
R
Raaj 已提交
737
          -DBUILD_python=OFF
G
gineshidalgo99 已提交
738 739 740 741 742
          -DBUILD_python_layer=OFF
          -DUSE_LEVELDB=OFF
          -DUSE_LMDB=OFF
          -DUSE_OPENCV=OFF)
          # -DOpenCV_DIR=${OpenCV_DIR})
743
      else (USE_MKL)
744
        ExternalProject_Add(openpose_lib
745 746
          SOURCE_DIR ${CAFFE_URL}
          PREFIX ${CAFFE_PREFIX}
G
gineshidalgo99 已提交
747
          CMAKE_ARGS -DCMAKE_INSTALL_PREFIX:PATH=<INSTALL_DIR>
G
gineshidalgo99 已提交
748
          -DUSE_CUDNN=${USE_CUDNN}
749
          -DCUDA_ARCH_NAME=${CUDA_ARCH}
750 751
          -DCUDA_ARCH_BIN=${CUDA_ARCH_BIN}
          -DCUDA_ARCH_PTX=${CUDA_ARCH_PTX}
G
gineshidalgo99 已提交
752 753 754
          -DCPU_ONLY=${CAFFE_CPU_ONLY}
          -DCMAKE_BUILD_TYPE=Release
          -DBUILD_docs=OFF
R
Raaj 已提交
755
          -DBUILD_python=OFF
G
gineshidalgo99 已提交
756 757 758 759 760
          -DBUILD_python_layer=OFF
          -DUSE_LEVELDB=OFF
          -DUSE_LMDB=OFF
          -DUSE_OPENCV=OFF)
          # -DOpenCV_DIR=${OpenCV_DIR})
761
      endif (USE_MKL)
B
Bikramjot Hanzra 已提交
762

763
      ExternalProject_Get_Property(openpose_lib install_dir)
764

B
Bikramjot Hanzra 已提交
765
      if (NOT Caffe_FOUND)
766
        add_custom_command(TARGET openpose_lib
B
Bikramjot Hanzra 已提交
767
            POST_BUILD
768
            COMMAND ${CMAKE_COMMAND} ${CMAKE_SOURCE_DIR}
B
Bikramjot Hanzra 已提交
769 770 771 772
            COMMAND $(MAKE)
            WORKING_DIRECTORY ${CMAKE_BINARY_DIR}
            COMMENT "Rerunning cmake after building Caffe submodule")
      endif (NOT Caffe_FOUND)
773 774 775 776

    endif (BUILD_CAFFE)

    if (NOT Caffe_FOUND AND NOT BUILD_CAFFE)
G
gineshidalgo99 已提交
777
      message(FATAL_ERROR "Caffe not found. Either turn on the BUILD_CAFFE option or specify the path of Caffe includes
778
        and libs using -DCaffe_INCLUDE_DIRS and -DCaffe_LIBS.")
779 780 781 782
    endif (NOT Caffe_FOUND AND NOT BUILD_CAFFE)

  endif (${DL_FRAMEWORK} MATCHES "CAFFE")

783
endif (UNIX OR APPLE)
B
Bikramjot Hanzra 已提交
784 785 786 787

### PROJECT INCLUDES
# Specify the include directories
include_directories(
788
  include
J
Jim Carroll 已提交
789
  ${Protobuf_INCLUDE_DIRS}
B
Bikramjot Hanzra 已提交
790 791 792
  ${GFLAGS_INCLUDE_DIR}
  ${GLOG_INCLUDE_DIR}
  ${OpenCV_INCLUDE_DIRS})
B
Bikramjot Hanzra 已提交
793

794 795 796 797 798 799 800 801
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)

802 803 804 805 806 807
# Calibration
if (NOT ${WITH_EIGEN} MATCHES "NONE")
  include_directories(
      ${EIGEN3_INCLUDE_DIRS})
endif (NOT ${WITH_EIGEN} MATCHES "NONE")

808 809 810 811 812
if (APPLE)
  include_directories(
      "/usr/local/opt/openblas/include")
endif (APPLE)

813
if (USE_MKL)
814 815 816 817
 include_directories(
     "${MKL_PATH}/include/")
endif (USE_MKL)

B
Bikramjot Hanzra 已提交
818 819 820 821 822
if (Caffe_FOUND)
  include_directories(
      ${Caffe_INCLUDE_DIRS})
endif (Caffe_FOUND)

G
gineshidalgo99 已提交
823 824
if (${GPU_MODE} MATCHES "CUDA")
  include_directories(
B
Bikramjot Hanzra 已提交
825
      ${CUDA_INCLUDE_DIRS})
826
elseif (${GPU_MODE} MATCHES "OPENCL")
827 828
  include_directories(
    ${OpenCL_INCLUDE_DIRS})
829
endif (${GPU_MODE} MATCHES "CUDA")
830
# 3D
831 832 833
if (WITH_3D_RENDERER)
  include_directories(${GLUT_INCLUDE_DIRS})
endif (WITH_3D_RENDERER)
834 835 836
if (WITH_CERES)
  include_directories(${CERES_INCLUDE_DIRS})
endif (WITH_CERES)
837
if (WITH_FLIR_CAMERA)
G
gineshidalgo99 已提交
838
  include_directories(SYSTEM ${SPINNAKER_INCLUDE_DIRS}) # To remove its warnings, equiv. to -isystem
839
endif (WITH_FLIR_CAMERA)
840
if (WITH_3D_ADAM_MODEL)
G
gineshidalgo99 已提交
841
                                    include_directories(include/adam) # TODO: TEMPORARY - TO BE REMOVED IN THE FUTURE
842 843 844 845
  include_directories(${CERES_INCLUDE_DIRS})
  include_directories(${EIGEN3_INCLUDE_DIRS})
  include_directories(${IGL_INCLUDE_DIRS})
  include_directories(${LIBIGL_INCLUDE_DIRS})
846
  include_directories(${GLUT_INCLUDE_DIRS} ${GLEW_INCLUDE_DIRS} ${OPENGL_INCLUDE_DIR})
847
endif (WITH_3D_ADAM_MODEL)
848
# Windows includes
849 850
if (WIN32)
  include_directories(
851
    # ${Boost_INCLUDE_DIRS}
852 853
    ${WINDOWS_INCLUDE_DIRS})
endif (WIN32)
B
Bikramjot Hanzra 已提交
854 855


856 857
### COLLECT ALL 3RD-PARTY LIBRARIES TO BE LINKED AGAINST
set(OpenPose_3rdparty_libraries ${OpenCV_LIBS} ${GLOG_LIBRARY})
858
if (UNIX OR APPLE)
859 860 861 862 863
  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})
864
endif (UNIX OR APPLE)
865 866 867 868 869 870 871 872 873 874 875
# 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 已提交
876 877 878 879 880 881 882 883 884
# 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)
885
# 3-D
886 887 888 889
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)
890
if (WITH_3D_RENDERER)
891
  set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries} ${GLUT_LIBRARY} ${OPENGL_LIBRARIES})
892 893 894 895 896 897 898
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 已提交
899
# Pthread
900
if (UNIX OR APPLE)
G
gineshidalgo99 已提交
901
  set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries} pthread)
902
endif (UNIX OR APPLE)
903 904 905 906

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


B
Bikramjot Hanzra 已提交
907 908 909 910 911 912 913
### ADD SUBDIRECTORIES
if (Caffe_FOUND)
  add_subdirectory(src)
  if (BUILD_EXAMPLES)
    add_subdirectory(examples)
  endif (BUILD_EXAMPLES)
endif (Caffe_FOUND)
B
Bikramjot Hanzra 已提交
914 915


G
gineshidalgo99 已提交
916
### DOWNLOAD MODELS
B
Bikramjot Hanzra 已提交
917 918 919 920 921
# 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 已提交
922

923 924 925
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 已提交
926
  5156d31f670511fce9b4e28b403f2939) # Body (COCO)
927
download_model("body (MPI)" ${DOWNLOAD_BODY_MPI_MODEL} pose/mpi/pose_iter_160000.caffemodel
G
gineshidalgo99 已提交
928
  2ca0990c7562bd7ae03f3f54afa96e00) # Body (MPI)
929
download_model("face" ${DOWNLOAD_FACE_MODEL} face/pose_iter_116000.caffemodel
G
gineshidalgo99 已提交
930
  e747180d728fa4e4418c465828384333) # Face
931
download_model("hand" ${DOWNLOAD_HAND_MODEL} hand/pose_iter_102000.caffemodel
G
gineshidalgo99 已提交
932
  a82cfc3fea7c62f159e11bd3674c1531) # Hand
B
Bikramjot Hanzra 已提交
933 934

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

R
Raaj 已提交
936

R
Raaj 已提交
937
### PYTHON
R
Raaj 已提交
938 939 940 941
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 已提交
942
    add_subdirectory(python)
R
Raaj 已提交
943 944 945
  elseif (UNIX OR APPLE)
    if (Caffe_FOUND)
      execute_process(COMMAND git submodule update --init ${CMAKE_SOURCE_DIR}/3rdparty/pybind11/)
946
      # execute_process(COMMAND git submodule update --init --recursive --remote) # This would initialize them all
R
Raaj 已提交
947 948 949 950 951 952 953 954
      add_subdirectory(3rdparty/pybind11)
      add_subdirectory(python)
    endif (Caffe_FOUND)
  else (WIN32)
    message(FATAL_ERROR "Unknown OS.")
  endif (WIN32)
endif (BUILD_PYTHON)

955

G
gineshidalgo99 已提交
956
### GENERATE DOCUMENTATION
957
if (UNIX OR APPLE)
958 959 960 961 962 963 964 965 966

  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 已提交
967 968 969 970
        COMMAND ${DOXYGEN_EXECUTABLE} ${DOXYGEN_FILE}
        WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/doc
        COMMENT "Generating API documentation with Doxygen"
        VERBATIM)
971
    else (DOXYGEN_FOUND)
972
      message(FATAL_ERROR "Doxygen needs to be installed to generate the doxygen documentation.")
973 974 975
    endif (DOXYGEN_FOUND)
  endif (BUILD_DOCS)

976
endif (UNIX OR APPLE)
977

B
bikz05 已提交
978

G
gineshidalgo99 已提交
979
### INSTALL
980
if (UNIX OR APPLE)
B
Bikramjot Hanzra 已提交
981 982 983 984 985 986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006
  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)
1007
endif (UNIX OR APPLE)