CMakeLists.txt 40.7 KB
Newer Older
B
bikz05 已提交
1 2
### VERSION INFO
set(OpenPose_VERSION_MAJOR 1)
G
Gines Hidalgo 已提交
3
set(OpenPose_VERSION_MINOR 5)
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)
H
henczati 已提交
34 35
cmake_minimum_required(VERSION 3.0 FATAL_ERROR) # required by cmake eigen finder to find locally installed versions (e.g., CUDA 9.2 support was added in eigen 3.3.5)
# cmake_minimum_required(VERSION 2.8.7 FATAL_ERROR) # min. cmake version recommended by Caffe
36 37


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


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


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

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

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

81 82 83 84 85 86 87 88 89
  # https://stackoverflow.com/questions/43697154/detect-current-cmake-version-using-cmake
  message(STATUS "CMAKE_VERSION = ${CMAKE_VERSION}")
  # Remove "minSizeRel" and "relWithDebInfo"
  if (${CMAKE_VERSION} VERSION_LESS "3.14.0")
    set(CMAKE_CONFIGURATION_TYPES Release Debug CACHE TYPE INTERNAL FORCE)
  else (${CMAKE_VERSION} VERSION_LESS "3.14.0")
    # But the above line gives warning on CMake >= 3.14, alternative
    set(CMAKE_CONFIGURATION_TYPES Release Debug CACHE STRING "" FORCE)
  endif (${CMAKE_VERSION} VERSION_LESS "3.14.0")
90

91 92 93 94 95 96 97 98 99
  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")
100

101 102 103
  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}")
104
elseif (UNIX)
G
gineshidalgo99 已提交
105
  # Turn on C++11
106 107
  add_definitions(-std=c++11)
  set(CMAKE_CXX_FLAGS_RELEASE "-O3")
108 109 110 111
elseif (APPLE)
  # Turn on C++11
  set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11")
  set(CMAKE_CXX_FLAGS_RELEASE "-O3")
112
endif (WIN32)
B
Bikramjot Hanzra 已提交
113 114

# C++ additional flags
G
gineshidalgo99 已提交
115
if (CMAKE_COMPILER_IS_GNUCXX)
G
Gines Hidalgo 已提交
116 117 118 119 120 121
  message(STATUS "GCC detected, adding compile flags")
  # set(OP_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fopenmp -Wpedantic -Wall -Wextra -Wfatal-errors")
  set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fopenmp")
  set(OP_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wpedantic -Wall -Wextra -Wfatal-errors")
  # set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS} -Wpedantic -Wall -Wextra -Wfatal-errors")
endif (CMAKE_COMPILER_IS_GNUCXX)
122

G
Gines Hidalgo 已提交
123 124 125 126 127 128 129 130 131 132 133 134 135 136 137
# Select the Enhanced Instruction Set
set(INSTRUCTION_SET NONE CACHE STRING "Enable Enhanced Instruction Set")
set_property(CACHE INSTRUCTION_SET PROPERTY STRINGS NONE AVX2)
if (${INSTRUCTION_SET} MATCHES "AVX2")
  add_definitions("-DWITH_AVX")
endif (${INSTRUCTION_SET} MATCHES "AVX2")
# Windows
if (WIN32)
  # Suboptions for Enhanced Instruction Set
  if (${INSTRUCTION_SET} MATCHES "AVX2")
    add_definitions("/arch:AVX2")
  endif (${INSTRUCTION_SET} MATCHES "AVX2")
# Ubuntu
else (WIN32) # if (CMAKE_COMPILER_IS_GNUCXX)
  if (${INSTRUCTION_SET} MATCHES "AVX2")
138
    set(SIMD_FLAGS "${SIMD_FLAGS} -mavx -march=native")
G
Gines Hidalgo 已提交
139
  endif (${INSTRUCTION_SET} MATCHES "AVX2")
G
gineshidalgo99 已提交
140
  message(STATUS "GCC detected, adding compile flags")
G
Gines Hidalgo 已提交
141 142 143 144 145 146 147 148 149 150
  set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${SIMD_FLAGS}")
endif (WIN32)
# if (CMAKE_COMPILER_IS_GNUCXX)
#   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")
#   endif (${INSTRUCTION_SET} MATCHES "SSE4")
#   if (${INSTRUCTION_SET} MATCHES "AVX2")
#     add_definitions("-DWITH_AVX")
151 152
#     set(SIMD_FLAGS "${SIMD_FLAGS} -mavx -march=native")
#   endif (${INSTRUCTION_SET} MATCHES "AVX2")
G
Gines Hidalgo 已提交
153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170
#   message(STATUS "GCC detected, adding compile flags")
#   set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${SIMD_FLAGS}")
# endif (CMAKE_COMPILER_IS_GNUCXX)
# if (WIN32)
#   # 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 已提交
171 172


G
gineshidalgo99 已提交
173
### PROJECT OPTIONS
B
Bikramjot Hanzra 已提交
174 175 176 177 178 179 180 181 182
# 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 已提交
183
    "DL_FRAMEWORK" ON)
184 185
  # OpenPose flags
  add_definitions(-DUSE_CAFFE)
B
Bikramjot Hanzra 已提交
186 187 188
endif (${DL_FRAMEWORK} MATCHES "CAFFE")

# Set the acceleration library
G
gineshidalgo99 已提交
189 190 191 192 193
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)
194 195 196 197
# 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 已提交
198
#   set_property(CACHE GPU_MODE PROPERTY STRINGS CUDA CPU_ONLY)
199 200 201 202
# elseif (OpenCL_FOUND)
#   set_property(CACHE GPU_MODE PROPERTY STRINGS OPENCL CPU_ONLY)
# else ()
#   set_property(CACHE GPU_MODE PROPERTY STRINGS CPU_ONLY)
203
# endif (CUDA_FOUND AND OpenCL_FOUND)
R
Raaj 已提交
204
set_property(CACHE GPU_MODE PROPERTY STRINGS CUDA OPENCL CPU_ONLY)
G
gineshidalgo99 已提交
205

206
# Look for CUDA
G
gineshidalgo99 已提交
207
set(CUDA_FOUND FALSE)
208 209 210 211 212
if (${GPU_MODE} MATCHES "CUDA")
  find_package(CUDA)
endif (${GPU_MODE} MATCHES "CUDA")
# Look for OpenCL
set(OpenCL_FOUND FALSE)
G
gineshidalgo99 已提交
213
set(CUDA_VERSION_MAJOR 0)
214 215 216 217
if (${GPU_MODE} MATCHES "OPENCL")
  find_package(OpenCL)
endif (${GPU_MODE} MATCHES "OPENCL")

218 219 220 221 222 223 224
# 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)
225
endif (CUDA_FOUND AND OpenCL_FOUND AND ${CUDA_VERSION_MAJOR} LESS 9)
226
# Handle desired GPU mode option
227 228 229
if (${GPU_MODE} MATCHES "CUDA")
  # OpenPose flags
  add_definitions(-DUSE_CUDA)
G
gineshidalgo99 已提交
230
  message(STATUS "Building with CUDA.")
231
elseif (${GPU_MODE} MATCHES "CPU_ONLY")
G
gineshidalgo99 已提交
232 233
  add_definitions(-DUSE_CPU_ONLY)
  message(STATUS "Building CPU Only.")
234 235
  # OpenPose flag for Caffe
  add_definitions(-DCPU_ONLY)
236 237 238 239 240
elseif (${GPU_MODE} MATCHES "OPENCL")
  # OpenPose flag for Caffe
  add_definitions(-DUSE_OPENCL)
  add_definitions(-DUSE_GREENTEA)
  message(STATUS "Building with OpenCL.")
241
endif (${GPU_MODE} MATCHES "CUDA")
G
gineshidalgo99 已提交
242 243 244 245 246

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

250 251 252 253 254 255
if (${USE_MKL})
  # OpenPose flags
  add_definitions(-DUSE_MKL)
  message(STATUS "Building with MKL support.")
endif (${USE_MKL})

256 257 258 259 260
# Set/disable profiler
if (PROFILER_ENABLED)
  add_definitions(-DPROFILER_ENABLED)
endif (PROFILER_ENABLED)

B
Bikramjot Hanzra 已提交
261 262 263 264 265 266 267 268 269 270
# 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")
271

G
gineshidalgo99 已提交
272
# Suboptions for OpenPose 3D Reconstruction module and demo
273
option(WITH_3D_RENDERER "Add OpenPose 3D renderer module (it requires FreeGLUT library)." OFF)
274
if (UNIX AND NOT APPLE)
G
gineshidalgo99 已提交
275
  option(WITH_CERES "Add Ceres support for advanced 3-D reconstruction." OFF)
276
endif (UNIX AND NOT APPLE)
277
option(WITH_FLIR_CAMERA "Add FLIR (formerly Point Grey) camera code (requires Spinnaker SDK already installed)." OFF)
G
gineshidalgo99 已提交
278
# option(WITH_3D_ADAM_MODEL "Add 3-D Adam model (requires OpenGL, Ceres, Eigen, OpenMP, FreeImage, GLEW, and IGL already installed)." OFF)
279

G
gineshidalgo99 已提交
280
# Faster GUI rendering
281 282 283
# 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 已提交
284

285
# Set the acceleration library
G
gineshidalgo99 已提交
286
if (WIN32 OR APPLE)
H
henczati 已提交
287 288
  set(WITH_EIGEN NONE CACHE STRING "Select the Eigen mode: NONE if not required, BUILD to let OpenPose download it, or FIND to let CMake find it (e.g., if you installed it manually).")
  set_property(CACHE WITH_EIGEN PROPERTY STRINGS NONE BUILD FIND)
289
elseif (UNIX AND NOT APPLE)
H
henczati 已提交
290 291
  set(WITH_EIGEN NONE CACHE STRING "Select the Eigen mode: NONE if not required, BUILD to let OpenPose download it, or FIND to let CMake find it (e.g., if you installed it manually or used `sudo apt-get install libeigen3-dev`).")
  set_property(CACHE WITH_EIGEN PROPERTY STRINGS NONE BUILD FIND)
G
gineshidalgo99 已提交
292
endif (WIN32 OR APPLE)
293

G
gineshidalgo99 已提交
294 295 296 297 298
# # 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 已提交
299
# Download the models
300 301 302
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 已提交
303
option(DOWNLOAD_FACE_MODEL "Download face model." ON)
304 305 306
option(DOWNLOAD_HAND_MODEL "Download hand model." ON)

# Asio
G
gineshidalgo99 已提交
307
# option(USE_ASIO "Include Asio header-only library." OFF)
B
Bikramjot Hanzra 已提交
308 309 310 311

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

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

B
bikz05 已提交
320
# Build as shared library
R
Raaj 已提交
321
option(BUILD_SHARED_LIBS "Build as shared lib." ON)
B
bikz05 已提交
322

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

G
gineshidalgo99 已提交
326
# Threads - Pthread
327
if (${GPU_MODE} MATCHES "OPENCL" OR (UNIX OR APPLE))
328 329
  unset(CMAKE_THREAD_LIBS_INIT CACHE)
  find_package(Threads)
330 331 332 333
endif (${GPU_MODE} MATCHES "OPENCL" OR (UNIX OR APPLE))

# Caffe OpenCL Boost Issue
if (APPLE)
R
Raaj 已提交
334
  if (${GPU_MODE} MATCHES "OPENCL" OR BUILD_PYTHON)
335 336 337 338 339
    unset(Boost_SYSTEM_LIBRARY CACHE)
    find_package(Boost COMPONENTS system REQUIRED)
  else ()
    set(Boost_SYSTEM_LIBRARY "")
  endif ()
G
gineshidalgo99 已提交
340
endif (APPLE)
B
Bikramjot Hanzra 已提交
341

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

G
gineshidalgo99 已提交
345 346
if (${GPU_MODE} MATCHES "CUDA")
  find_package(CUDA)
347
endif (${GPU_MODE} MATCHES "CUDA")
G
gineshidalgo99 已提交
348

349
# Adding 3D
G
gineshidalgo99 已提交
350 351 352 353
if (WITH_OPENCV_WITH_OPENGL)
  # OpenPose flags
  add_definitions(-DUSE_OPENCV_WITH_OPENGL)
endif (WITH_OPENCV_WITH_OPENGL)
354
if (WITH_3D_RENDERER)
355
  # OpenPose flags
356
  add_definitions(-DUSE_3D_RENDERER)
357
endif (WITH_3D_RENDERER)
358
if (WITH_CERES)
359
  add_definitions(-DUSE_CERES)
360
endif (WITH_CERES)
361 362
if (WITH_FLIR_CAMERA)
  # OpenPose flags
363
  add_definitions(-DUSE_FLIR_CAMERA)
364
endif (WITH_FLIR_CAMERA)
365 366 367 368
if (WITH_3D_ADAM_MODEL)
  # OpenPose flags
  add_definitions(-DUSE_3D_ADAM_MODEL)
endif (WITH_3D_ADAM_MODEL)
369

G
gineshidalgo99 已提交
370 371 372
# Adding tracking
if (WITH_TRACKING)
  # OpenPose flags
373
  add_definitions(-DUSE_TRACKING)
G
gineshidalgo99 已提交
374 375
endif (WITH_TRACKING)

G
gineshidalgo99 已提交
376 377 378 379 380 381
# Unity
if (BUILD_UNITY_SUPPORT)
  # OpenPose flags
  add_definitions(-DUSE_UNITY_SUPPORT)
endif (BUILD_UNITY_SUPPORT)

382 383 384 385 386 387 388 389 390
# 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
391
  add_definitions(-DUSE_EIGEN)
392 393 394 395 396 397 398 399
  # 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/")
H
henczati 已提交
400 401
  # Leave to CMake to find it
  elseif (${WITH_EIGEN} MATCHES "FIND")
402
    # Eigen
H
henczati 已提交
403 404 405 406 407 408 409 410 411
    # # Option a) apt-get
    # find_package(PkgConfig)
    # pkg_check_modules(EIGEN3 REQUIRED eigen3)
    # # Option b) Find it (including apt-get)
    if (CUDA_VERSION_STRING STRGREATER "9.0")
      find_package(Eigen3 3.3.5 REQUIRED NO_MODULE)
    else (CUDA_VERSION_STRING STRGREATER "9.0")
      find_package(Eigen3 3 REQUIRED NO_MODULE)
    endif (CUDA_VERSION_STRING STRGREATER "9.0")
412 413 414
  endif (${WITH_EIGEN} MATCHES "BUILD")
endif (${WITH_EIGEN} MATCHES "NONE")

415
if (UNIX OR APPLE)
G
gineshidalgo99 已提交
416 417 418
  if (${GPU_MODE} MATCHES "CUDA")
    include(cmake/Cuda.cmake)
    find_package(CuDNN)
419
  endif (${GPU_MODE} MATCHES "CUDA")
J
Jim Carroll 已提交
420 421 422 423
  find_package(GFlags) # For Caffe and OpenPose
  find_package(Glog) # For Caffe
  find_package(Protobuf REQUIRED) # For Caffe

B
Bikramjot Hanzra 已提交
424 425
  if (OpenCV_CONFIG_FILE)
    include (${OpenCV_CONFIG_FILE})
J
Jim Carroll 已提交
426 427 428
  # Allow explicitly setting the OpenCV includes and libs
  elseif (OpenCV_INCLUDE_DIRS AND OpenCV_LIBS)
    set(OpenCV_FOUND 1)
B
Bikramjot Hanzra 已提交
429 430 431
  elseif (OpenCV_INCLUDE_DIRS AND OpenCV_LIBS_DIR)
    file(GLOB_RECURSE OpenCV_LIBS "${OpenCV_LIBS_DIR}*.so")
    set(OpenCV_FOUND 1)
432
  else (OpenCV_CONFIG_FILE)
B
Bikramjot Hanzra 已提交
433 434
    find_package(OpenCV)
  endif (OpenCV_CONFIG_FILE)
435

436
  # 3D
437
  if (WITH_3D_RENDERER)
H
henczati 已提交
438 439
    # OpenGL
    find_package(OpenGL REQUIRED)
440 441
    # GLUT
    find_package(GLUT REQUIRED)
442 443
    # OpenGL
    find_package(OpenGL REQUIRED)
444
  endif (WITH_3D_RENDERER)
445 446 447 448
  if (WITH_CERES)
    # Eigen + Ceres
    find_package(Ceres REQUIRED COMPONENTS SuiteSparse)
  endif (WITH_CERES)
449
  if (WITH_FLIR_CAMERA)
450 451 452
    # Spinnaker
    find_package(Spinnaker)
    if (NOT SPINNAKER_FOUND)
453 454
      message(FATAL_ERROR "Spinnaker not found. Either turn off the `WITH_FLIR_CAMERA` option or specify the path to
        the Spinnaker includes and libs.")
455
    endif (NOT SPINNAKER_FOUND)
456
  endif (WITH_FLIR_CAMERA)
457
  if (WITH_3D_ADAM_MODEL)
G
gineshidalgo99 已提交
458 459
    if (NOT WITH_3D_RENDERER)
      message(FATAL_ERROR "WITH_3D_RENDERER is required if WITH_3D_ADAM_MODEL is enabled.")
460
    endif (NOT WITH_3D_RENDERER)
H
henczati 已提交
461 462 463 464 465 466 467 468 469 470
    # Eigen
    # # Option a) apt-get
    # find_package(PkgConfig)
    # pkg_check_modules(EIGEN3 REQUIRED eigen3)
    # # Option b) Find it (including apt-get)
    if (CUDA_VERSION_STRING STRGREATER "9.0")
      find_package(Eigen3 3.3.5 REQUIRED NO_MODULE)
    else (CUDA_VERSION_STRING STRGREATER "9.0")
      find_package(Eigen3 3 REQUIRED NO_MODULE)
    endif (CUDA_VERSION_STRING STRGREATER "9.0")
G
gineshidalgo99 已提交
471 472
    # Others: sudo apt-get install libglm-dev
    # http://ceres-solver.org
473
    find_package(Ceres REQUIRED COMPONENTS SuiteSparse)
G
gineshidalgo99 已提交
474
    # sudo apt-get install libglew-dev
475
    find_package(GLEW REQUIRED)
G
gineshidalgo99 已提交
476 477 478 479
    # 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
480 481
    find_package(LIBIGL REQUIRED)
    find_package(OpenMP REQUIRED)
G
gineshidalgo99 已提交
482 483
    # Only adam/renderer::Renderer::IdleSaveImage() uses it. Make dependency optional in hand_model
    # FIND_LIBRARY(FREE_IMAGE_LIBRARY NAMES libfreeimage.so)
484 485 486 487
    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)
488

G
gineshidalgo99 已提交
489 490 491 492
  # OpenMP
  if (${GPU_MODE} MATCHES "CPU_ONLY")
    find_package(OpenMP)
    if (OPENMP_FOUND)
G
gineshidalgo99 已提交
493 494 495
      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}")
496 497
    endif (OPENMP_FOUND)
  endif (${GPU_MODE} MATCHES "CPU_ONLY")
G
gineshidalgo99 已提交
498 499 500 501 502 503 504

  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.")
505
      execute_process(COMMAND cat install_cuda.sh WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/scripts/ubuntu)
G
gineshidalgo99 已提交
506 507 508 509 510
      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.")
511
      execute_process(COMMAND cat install_cudnn.sh WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/scripts/ubuntu)
G
gineshidalgo99 已提交
512 513
      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)
514
  endif (${GPU_MODE} MATCHES "CUDA")
515 516

  if (NOT GLOG_FOUND)
G
gineshidalgo99 已提交
517 518
    message(FATAL_ERROR "Glog not found. Install Glog from the command line using the command(s) -\
      sudo apt-get install libgoogle-glog-dev")
519 520 521
  endif (NOT GLOG_FOUND)

  if (NOT GFLAGS_FOUND)
G
gineshidalgo99 已提交
522
    message(FATAL_ERROR "GFlags not found. Install GFlags from the command line using the command(s) --\
523 524 525 526 527 528 529 530
      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)

531
endif (UNIX OR APPLE)
532 533 534

if (WIN32)

G
gineshidalgo99 已提交
535
  if ("${CMAKE_VERSION}" VERSION_GREATER 3.6.3)
536
    set_property(DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR} PROPERTY VS_STARTUP_PROJECT OpenPoseDemo)
G
gineshidalgo99 已提交
537
  endif ("${CMAKE_VERSION}" VERSION_GREATER 3.6.3)
538 539 540 541 542 543 544 545 546

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

547
  # Download required zip files
548 549 550
  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)
551 552
  if (${GPU_MODE} MATCHES "OPENCL")
    download_zip("caffe_opencl_2018_02_13.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} 3ac3e1acf5ee6a4e57920be73053067a)
553 554
  elseif (${GPU_MODE} MATCHES "CPU_ONLY")
    download_zip("caffe_cpu_2018_05_27.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} 87E8401B6DFBAC5B8E909DD20E3B3390)
555
  else (${GPU_MODE} MATCHES "OPENCL")
556
    # download_zip("caffe_2019_03_12.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} 859a592310f0928fd4f40da1456a217f)
557
    download_zip("caffe_15_2019_05_16.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} f2d8beee9e4959fa419dcb54d507107c)
558
  endif (${GPU_MODE} MATCHES "OPENCL")
559
  if (WITH_3D_RENDERER)
560
    download_zip("freeglut_2018_01_14.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} BB182187285E06880F0EDE3A39530091)
561
  endif (WITH_3D_RENDERER)
562 563
  message(STATUS "Windows dependencies downloaded.")

564 565 566
  # 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)
567 568 569 570
  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)
571

572 573 574
  # If OpenPose builds it
  if (BUILD_CAFFE)
    unset(Caffe_INCLUDE_DIRS CACHE)
575 576
    unset(Caffe_LIB CACHE)
    unset(Caffe_Proto_LIB CACHE)
577 578 579 580
  endif (BUILD_CAFFE)
  # OpenCL
  if (${GPU_MODE} MATCHES "OPENCL")
    set(VCXPROJ_FILE_GPU_MODE "_CL")
581 582
    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)
583
  # CPU & CUDA
584
  else (${GPU_MODE} MATCHES "OPENCL")
585 586 587 588 589 590 591 592 593 594 595
    # 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")
596
  endif (${GPU_MODE} MATCHES "OPENCL")
R
Raaj 已提交
597 598
  # Boost DepCopy over required DLL F
  if (${GPU_MODE} MATCHES "CPU_ONLY" OR ${GPU_MODE} MATCHES "OPENCL" OR BUILD_PYTHON)
599 600 601 602
      find_library(BOOST_SYSTEM_LIB_RELEASE boost_system-vc141-mt-x64-1_69 HINTS ${FIND_LIB_PREFIX}/caffe3rdparty/lib)
      find_library(BOOST_SYSTEM_LIB_DEBUG boost_system-vc141-mt-gd-x64-1_69 HINTS ${FIND_LIB_PREFIX}/caffe3rdparty/lib)
      find_library(BOOST_FILESYSTEM_LIB_RELEASE boost_filesystem-vc141-mt-x64-1_69 HINTS ${FIND_LIB_PREFIX}/caffe3rdparty/lib)
      find_library(BOOST_FILESYSTEM_LIB_DEBUG boost_filesystem-vc141-mt-gd-x64-1_69 HINTS ${FIND_LIB_PREFIX}/caffe3rdparty/lib)
R
Raaj 已提交
603 604 605 606 607 608
  else ()
      set(BOOST_SYSTEM_LIB_RELEASE "")
      set(BOOST_SYSTEM_LIB_DEBUG "")
      set(BOOST_FILESYSTEM_LIB_RELEASE "")
      set(BOOST_FILESYSTEM_LIB_DEBUG "")
  endif ()
609
  if (WITH_3D_RENDERER)
610 611
    find_library(GLUT_LIBRARY freeglut HINTS ${FIND_LIB_PREFIX}/freeglut/lib)
    message(STATUS "\${GLUT_LIBRARY} = ${GLUT_LIBRARY}")
612 613 614 615
  endif (WITH_3D_RENDERER)
  if (WITH_FLIR_CAMERA)
    find_library(SPINNAKER_LIB spinnaker_v140 HINTS ${FIND_LIB_PREFIX}/spinnaker/lib)
  endif (WITH_FLIR_CAMERA)
616 617
  set(Caffe_LIBS ${Caffe_LIB};${Caffe_Proto_LIB})
  set(OpenCV_INCLUDE_DIRS "3rdparty/windows/opencv/include")
618
  # OpenCL
619 620 621
  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")
622
  # CUDA and CPU
623
  else (${GPU_MODE} MATCHES "OPENCL")
624 625 626 627 628 629 630
    # 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")
631
  endif (${GPU_MODE} MATCHES "OPENCL")
632
  # set(Boost_INCLUDE_DIRS "3rdparty/windows/caffe3rdparty/include/boost-1_61") # Added in WINDOWS_INCLUDE_DIRS
633
  set(WINDOWS_INCLUDE_DIRS "3rdparty/windows/caffe3rdparty/include")
634
  if (WITH_3D_RENDERER)
635
    set(GLUT_INCLUDE_DIRS "3rdparty/windows/freeglut/include")
636 637 638 639
  endif (WITH_3D_RENDERER)
  if (WITH_FLIR_CAMERA)
    set(SPINNAKER_INCLUDE_DIRS "3rdparty/windows/spinnaker/include")
  endif (WITH_FLIR_CAMERA)
B
Bikramjot Hanzra 已提交
640
  set(Caffe_FOUND 1)
R
Raaj 已提交
641 642 643

  # Build DLL Must be on if Build Python is on
  if (BUILD_PYTHON)
G
gineshidalgo99 已提交
644 645
    if (NOT BUILD_BIN_FOLDER)
      message(FATAL_ERROR "BUILD_BIN_FOLDER must be turned on to as well to build python library")
G
gineshidalgo99 已提交
646 647
    endif (NOT BUILD_BIN_FOLDER)
  endif (BUILD_PYTHON)
R
Raaj 已提交
648 649

  # Auto copy DLLs
G
gineshidalgo99 已提交
650 651 652
  if (BUILD_BIN_FOLDER)
    # Locate DLLs
    # Caffe DLLs
R
Raaj 已提交
653 654 655 656 657 658 659
    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 已提交
660
    # Caffe 3rd-party DLLs
661
    file(GLOB CAFFE_3RD_PARTY_DLL "${CMAKE_SOURCE_DIR}/3rdparty/windows/caffe3rdparty/lib/*.dll")
G
gineshidalgo99 已提交
662
    # OpenCV DLLs
663
    file(GLOB OPENCV_DLL "${CMAKE_SOURCE_DIR}/3rdparty/windows/opencv/x64/vc${VS_VERSION}/bin/*.dll")
664 665 666 667
    # 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 已提交
668 669 670 671 672
    # 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})
673 674 675
    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 已提交
676
  endif (BUILD_BIN_FOLDER)
R
Raaj 已提交
677

678
endif (WIN32)
B
Bikramjot Hanzra 已提交
679 680


G
gineshidalgo99 已提交
681
### CAFFE
682
if (UNIX OR APPLE)
B
Bikramjot Hanzra 已提交
683

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

686 687 688 689 690 691
    # 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 已提交
692

693 694
    # Else build from scratch
    if (BUILD_CAFFE)
B
Bikramjot Hanzra 已提交
695

696 697
      # Download Caffe
      message(STATUS "Caffe will be downloaded from source now. NOTE: This process might take several minutes depending
G
gineshidalgo99 已提交
698 699 700 701 702 703
        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 已提交
704
        execute_process(COMMAND git submodule update --init ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
705
        # execute_process(COMMAND git submodule update --init --recursive --remote) # This would initialize them all
706
      else (CAFFE_DIR_VALID_LENGTH EQUAL 0)
G
gineshidalgo99 已提交
707
        message(STATUS "Caffe has already been downloaded.")
708
      endif (CAFFE_DIR_VALID_LENGTH EQUAL 0)
G
gineshidalgo99 已提交
709 710 711 712

      # Build Process
      set(CAFFE_CPU_ONLY OFF)
      if (${GPU_MODE} MATCHES "CUDA")
G
gineshidalgo99 已提交
713 714
        # 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 已提交
715 716
      elseif (${GPU_MODE} MATCHES "CPU_ONLY")
        if (USE_MKL)
G
gineshidalgo99 已提交
717 718
          #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 已提交
719 720
          execute_process(COMMAND sh prepare_mkl.sh WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/3rdparty/caffe/external/mkl
              OUTPUT_VARIABLE rv)
G
gineshidalgo99 已提交
721 722 723
          set( MLIST ${rv} )
          separate_arguments(MLIST)
          list(GET MLIST 0 MKL_PATH)
724
          message(STATUS ${MKL_PATH})
G
gineshidalgo99 已提交
725 726 727 728 729
          file(GLOB MKL_SO
            "${MKL_PATH}lib/*"
            )
          file(COPY ${MKL_SO} DESTINATION ${CMAKE_BINARY_DIR}/caffe)

G
gineshidalgo99 已提交
730 731 732 733 734 735 736 737 738 739 740 741
          # 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 已提交
742
          set(MKL_LIBS
G
gineshidalgo99 已提交
743
            #"${CMAKE_BINARY_DIR}/caffe/libmlsl.so"
G
gineshidalgo99 已提交
744 745 746 747
            "${CMAKE_BINARY_DIR}/caffe/libiomp5.so"
            "${CMAKE_BINARY_DIR}/caffe/libmklml_intel.so"
            "${CMAKE_BINARY_DIR}/caffe/lib/libmkldnn.so"
            )
748
        else (USE_MKL)
G
gineshidalgo99 已提交
749 750
          # 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)
751
        endif (USE_MKL)
G
gineshidalgo99 已提交
752 753
        set(CAFFE_CPU_ONLY ON)
        set(USE_CUDNN OFF)
754
      elseif (${GPU_MODE} MATCHES "OPENCL")
R
Raaj 已提交
755
        execute_process(COMMAND git checkout fe2a1102 WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
756
        set(USE_CUDNN OFF)
757
      endif (${GPU_MODE} MATCHES "CUDA")
G
gineshidalgo99 已提交
758

759 760
      # Build Caffe
      message(STATUS "Caffe will be built from source now.")
B
Bikramjot Hanzra 已提交
761
      find_package(Caffe)
762 763 764
      include(ExternalProject)
      set(CAFFE_PREFIX caffe)
      set(CAFFE_URL ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
B
Bikramjot Hanzra 已提交
765

G
gineshidalgo99 已提交
766 767
      # One for Intel Branch and one for Master
      if (USE_MKL)
768
        ExternalProject_Add(openpose_lib
G
gineshidalgo99 已提交
769 770 771 772
          SOURCE_DIR ${CAFFE_URL}
          PREFIX ${CAFFE_PREFIX}
          CMAKE_ARGS -DCMAKE_INSTALL_PREFIX:PATH=<INSTALL_DIR>
          -DMKLDNN_INSTALL_DIR:PATH=<INSTALL_DIR>
773
          -DUSE_MKL2017_AS_DEFAULT_ENGINE=${CAFFE_CPU_ONLY}
G
gineshidalgo99 已提交
774
          -DUSE_CUDNN=${USE_CUDNN}
775
          -DCUDA_ARCH_NAME=${CUDA_ARCH}
776 777
          -DCUDA_ARCH_BIN=${CUDA_ARCH_BIN}
          -DCUDA_ARCH_PTX=${CUDA_ARCH_PTX}
G
gineshidalgo99 已提交
778 779 780
          -DCPU_ONLY=${CAFFE_CPU_ONLY}
          -DCMAKE_BUILD_TYPE=Release
          -DBUILD_docs=OFF
R
Raaj 已提交
781
          -DBUILD_python=OFF
G
gineshidalgo99 已提交
782 783 784 785 786
          -DBUILD_python_layer=OFF
          -DUSE_LEVELDB=OFF
          -DUSE_LMDB=OFF
          -DUSE_OPENCV=OFF)
          # -DOpenCV_DIR=${OpenCV_DIR})
787
      else (USE_MKL)
788
        ExternalProject_Add(openpose_lib
789 790
          SOURCE_DIR ${CAFFE_URL}
          PREFIX ${CAFFE_PREFIX}
G
gineshidalgo99 已提交
791
          CMAKE_ARGS -DCMAKE_INSTALL_PREFIX:PATH=<INSTALL_DIR>
G
gineshidalgo99 已提交
792
          -DUSE_CUDNN=${USE_CUDNN}
793
          -DCUDA_ARCH_NAME=${CUDA_ARCH}
794 795
          -DCUDA_ARCH_BIN=${CUDA_ARCH_BIN}
          -DCUDA_ARCH_PTX=${CUDA_ARCH_PTX}
G
gineshidalgo99 已提交
796 797 798
          -DCPU_ONLY=${CAFFE_CPU_ONLY}
          -DCMAKE_BUILD_TYPE=Release
          -DBUILD_docs=OFF
R
Raaj 已提交
799
          -DBUILD_python=OFF
G
gineshidalgo99 已提交
800 801 802 803 804
          -DBUILD_python_layer=OFF
          -DUSE_LEVELDB=OFF
          -DUSE_LMDB=OFF
          -DUSE_OPENCV=OFF)
          # -DOpenCV_DIR=${OpenCV_DIR})
805
      endif (USE_MKL)
B
Bikramjot Hanzra 已提交
806

807
      ExternalProject_Get_Property(openpose_lib install_dir)
808

B
Bikramjot Hanzra 已提交
809
      if (NOT Caffe_FOUND)
810
        add_custom_command(TARGET openpose_lib
B
Bikramjot Hanzra 已提交
811
            POST_BUILD
812
            COMMAND ${CMAKE_COMMAND} ${CMAKE_SOURCE_DIR}
B
Bikramjot Hanzra 已提交
813 814 815 816
            COMMAND $(MAKE)
            WORKING_DIRECTORY ${CMAKE_BINARY_DIR}
            COMMENT "Rerunning cmake after building Caffe submodule")
      endif (NOT Caffe_FOUND)
817 818 819 820

    endif (BUILD_CAFFE)

    if (NOT Caffe_FOUND AND NOT BUILD_CAFFE)
G
gineshidalgo99 已提交
821
      message(FATAL_ERROR "Caffe not found. Either turn on the BUILD_CAFFE option or specify the path of Caffe includes
822
        and libs using -DCaffe_INCLUDE_DIRS and -DCaffe_LIBS.")
823 824 825 826
    endif (NOT Caffe_FOUND AND NOT BUILD_CAFFE)

  endif (${DL_FRAMEWORK} MATCHES "CAFFE")

827
endif (UNIX OR APPLE)
B
Bikramjot Hanzra 已提交
828 829 830 831

### PROJECT INCLUDES
# Specify the include directories
include_directories(
832
  include
J
Jim Carroll 已提交
833
  ${Protobuf_INCLUDE_DIRS}
B
Bikramjot Hanzra 已提交
834 835 836
  ${GFLAGS_INCLUDE_DIR}
  ${GLOG_INCLUDE_DIR}
  ${OpenCV_INCLUDE_DIRS})
B
Bikramjot Hanzra 已提交
837

838 839 840 841 842 843 844 845
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)

846 847 848 849
# Calibration
if (NOT ${WITH_EIGEN} MATCHES "NONE")
  include_directories(
      ${EIGEN3_INCLUDE_DIRS})
H
henczati 已提交
850
  set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries} Eigen3::Eigen)
851 852
endif (NOT ${WITH_EIGEN} MATCHES "NONE")

853 854 855 856 857
if (APPLE)
  include_directories(
      "/usr/local/opt/openblas/include")
endif (APPLE)

858
if (USE_MKL)
859 860 861 862
 include_directories(
     "${MKL_PATH}/include/")
endif (USE_MKL)

B
Bikramjot Hanzra 已提交
863 864 865 866 867
if (Caffe_FOUND)
  include_directories(
      ${Caffe_INCLUDE_DIRS})
endif (Caffe_FOUND)

G
gineshidalgo99 已提交
868 869
if (${GPU_MODE} MATCHES "CUDA")
  include_directories(
B
Bikramjot Hanzra 已提交
870
      ${CUDA_INCLUDE_DIRS})
871
elseif (${GPU_MODE} MATCHES "OPENCL")
872 873
  include_directories(
    ${OpenCL_INCLUDE_DIRS})
874
endif (${GPU_MODE} MATCHES "CUDA")
875
# 3D
876 877 878
if (WITH_3D_RENDERER)
  include_directories(${GLUT_INCLUDE_DIRS})
endif (WITH_3D_RENDERER)
879 880 881
if (WITH_CERES)
  include_directories(${CERES_INCLUDE_DIRS})
endif (WITH_CERES)
882
if (WITH_FLIR_CAMERA)
G
gineshidalgo99 已提交
883
  include_directories(SYSTEM ${SPINNAKER_INCLUDE_DIRS}) # To remove its warnings, equiv. to -isystem
884
endif (WITH_FLIR_CAMERA)
885
if (WITH_3D_ADAM_MODEL)
G
gineshidalgo99 已提交
886
                                    include_directories(include/adam) # TODO: TEMPORARY - TO BE REMOVED IN THE FUTURE
887 888
  include_directories(${CERES_INCLUDE_DIRS})
  include_directories(${EIGEN3_INCLUDE_DIRS})
H
henczati 已提交
889
  set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries} Eigen3::Eigen)
890 891
  include_directories(${IGL_INCLUDE_DIRS})
  include_directories(${LIBIGL_INCLUDE_DIRS})
H
henczati 已提交
892
  include_directories(${OPENGL_INCLUDE_DIR} ${GLUT_INCLUDE_DIRS} ${GLEW_INCLUDE_DIRS})
893
endif (WITH_3D_ADAM_MODEL)
894
# Windows includes
895 896
if (WIN32)
  include_directories(
897
    # ${Boost_INCLUDE_DIRS}
898 899
    ${WINDOWS_INCLUDE_DIRS})
endif (WIN32)
B
Bikramjot Hanzra 已提交
900 901


902 903
### COLLECT ALL 3RD-PARTY LIBRARIES TO BE LINKED AGAINST
set(OpenPose_3rdparty_libraries ${OpenCV_LIBS} ${GLOG_LIBRARY})
904
if (UNIX OR APPLE)
905 906 907 908 909
  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})
910
endif (UNIX OR APPLE)
911 912 913 914 915 916 917 918 919 920 921
# 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 已提交
922 923 924 925 926 927 928 929 930
# 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)
931
# 3-D
932 933 934 935
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)
936
if (WITH_3D_RENDERER)
937
  set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries} ${GLUT_LIBRARY} ${OPENGL_LIBRARIES})
938 939 940 941 942 943 944
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 已提交
945
# Pthread
946
if (UNIX OR APPLE)
G
gineshidalgo99 已提交
947
  set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries} pthread)
948
endif (UNIX OR APPLE)
949 950 951 952

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


B
Bikramjot Hanzra 已提交
953 954 955 956 957 958 959
### ADD SUBDIRECTORIES
if (Caffe_FOUND)
  add_subdirectory(src)
  if (BUILD_EXAMPLES)
    add_subdirectory(examples)
  endif (BUILD_EXAMPLES)
endif (Caffe_FOUND)
B
Bikramjot Hanzra 已提交
960 961


G
gineshidalgo99 已提交
962
### DOWNLOAD MODELS
B
Bikramjot Hanzra 已提交
963 964 965 966 967
# 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 已提交
968

969 970 971
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 已提交
972
  5156d31f670511fce9b4e28b403f2939) # Body (COCO)
973
download_model("body (MPI)" ${DOWNLOAD_BODY_MPI_MODEL} pose/mpi/pose_iter_160000.caffemodel
G
gineshidalgo99 已提交
974
  2ca0990c7562bd7ae03f3f54afa96e00) # Body (MPI)
975
download_model("face" ${DOWNLOAD_FACE_MODEL} face/pose_iter_116000.caffemodel
G
gineshidalgo99 已提交
976
  e747180d728fa4e4418c465828384333) # Face
977
download_model("hand" ${DOWNLOAD_HAND_MODEL} hand/pose_iter_102000.caffemodel
G
gineshidalgo99 已提交
978
  a82cfc3fea7c62f159e11bd3674c1531) # Hand
B
Bikramjot Hanzra 已提交
979 980

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

R
Raaj 已提交
982

R
Raaj 已提交
983
### PYTHON
R
Raaj 已提交
984 985 986 987
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 已提交
988
    add_subdirectory(python)
R
Raaj 已提交
989 990 991
  elseif (UNIX OR APPLE)
    if (Caffe_FOUND)
      execute_process(COMMAND git submodule update --init ${CMAKE_SOURCE_DIR}/3rdparty/pybind11/)
992
      # execute_process(COMMAND git submodule update --init --recursive --remote) # This would initialize them all
R
Raaj 已提交
993 994 995 996 997 998 999 1000
      add_subdirectory(3rdparty/pybind11)
      add_subdirectory(python)
    endif (Caffe_FOUND)
  else (WIN32)
    message(FATAL_ERROR "Unknown OS.")
  endif (WIN32)
endif (BUILD_PYTHON)

1001

G
gineshidalgo99 已提交
1002
### GENERATE DOCUMENTATION
1003
if (UNIX OR APPLE)
1004 1005 1006 1007 1008 1009 1010 1011 1012

  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 已提交
1013 1014 1015 1016
        COMMAND ${DOXYGEN_EXECUTABLE} ${DOXYGEN_FILE}
        WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/doc
        COMMENT "Generating API documentation with Doxygen"
        VERBATIM)
1017
    else (DOXYGEN_FOUND)
1018
      message(FATAL_ERROR "Doxygen needs to be installed to generate the doxygen documentation.")
1019 1020 1021
    endif (DOXYGEN_FOUND)
  endif (BUILD_DOCS)

1022
endif (UNIX OR APPLE)
1023

B
bikz05 已提交
1024

G
gineshidalgo99 已提交
1025
### INSTALL
1026
if (UNIX OR APPLE)
B
Bikramjot Hanzra 已提交
1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052
  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)
1053
endif (UNIX OR APPLE)