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

7 8 9 10 11 12 13

### CMAKE HEADERS
cmake_policy(SET CMP0048 NEW)
project(OpenPose VERSION ${OpenPose_VERSION})
cmake_minimum_required(VERSION 2.8.7 FATAL_ERROR) # min. cmake version recommended by Caffe


B
Bikramjot Hanzra 已提交
14 15
### FIND THE OS
if (WIN32) # OS is Windows
16 17
  message(STATUS "Windows OS support currently is experimental.")
  message(STATUS "Feel free to send us any suggestion or bug you encounter.")
B
Bikramjot Hanzra 已提交
18 19 20
elseif (APPLE) # OS is Apple
  message(FATAL_ERROR "Apple OS is not currently supported.")
elseif (UNIX AND NOT APPLE) # OS is a Linux distribution (it assumes Ubuntu)
G
gineshidalgo99 已提交
21
  set(EXECUTE_COMMAND lsb_release -rs)
B
Bikramjot Hanzra 已提交
22 23 24 25 26
  execute_process(COMMAND ${EXECUTE_COMMAND} OUTPUT_VARIABLE UBUNTU_VERSION)
  string(SUBSTRING ${UBUNTU_VERSION} 0 2 UBUNTU_MAJOR_VERSION)
  if (${UBUNTU_MAJOR_VERSION} MATCHES "16") # if 16
    set(IS_UBUNTU_16 TRUE)
    set(IS_UBUNTU_14_OR_LESS FALSE)
27
  else (${UBUNTU_MAJOR_VERSION} MATCHES "16") # if 14 or less
B
Bikramjot Hanzra 已提交
28 29 30 31 32 33
    set(IS_UBUNTU_16 FALSE)
    set(IS_UBUNTU_14_OR_LESS TRUE)
  endif (${UBUNTU_MAJOR_VERSION} MATCHES "16")
endif ()


G
gineshidalgo99 已提交
34
### FLAGS
35
if (WIN32)
G
gineshidalgo99 已提交
36
  # TODO -- Makeshift solution -- This prevents rerunning build again
37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61
  # https://gitlab.kitware.com/cmake/cmake/issues/16783
  set(CMAKE_SUPPRESS_REGENERATION ON)

  string (REPLACE "/D_WINDOWS" "" CMAKE_CXX_FLAGS ${CMAKE_CXX_FLAGS})
  string (REPLACE "/DWIN32" "" CMAKE_CXX_FLAGS ${CMAKE_CXX_FLAGS})

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

  set(CMAKE_CONFIGURATION_TYPES Debug Release CACHE TYPE INTERNAL FORCE)
  
  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")
  
  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}")
elseif (UNIX AND NOT APPLE)
G
gineshidalgo99 已提交
62
  # Turn on C++11
63 64 65
  add_definitions(-std=c++11)
  set(CMAKE_CXX_FLAGS_RELEASE "-O3")
endif (WIN32)
B
Bikramjot Hanzra 已提交
66 67

# C++ additional flags
G
gineshidalgo99 已提交
68 69 70 71
if (CMAKE_COMPILER_IS_GNUCXX)
  message(STATUS "GCC detected, adding compile flags")
  set(OP_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fopenmp -Wpedantic -Wall -Wextra -Wfatal-errors")
endif (CMAKE_COMPILER_IS_GNUCXX)
B
Bikramjot Hanzra 已提交
72 73


G
gineshidalgo99 已提交
74
### PROJECT OPTIONS
B
Bikramjot Hanzra 已提交
75 76 77 78 79 80 81 82 83
# 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 已提交
84
    "DL_FRAMEWORK" ON)
85 86
  # OpenPose flags
  add_definitions(-DUSE_CAFFE)
B
Bikramjot Hanzra 已提交
87 88
endif (${DL_FRAMEWORK} MATCHES "CAFFE")

89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107
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 已提交
108
# Set the acceleration library
G
gineshidalgo99 已提交
109
set(GPU_MODE CUDA CACHE STRING "Select the acceleration GPU library or CPU otherwise.")
G
gineshidalgo99 已提交
110 111 112 113 114 115 116 117
# set_property(CACHE GPU_MODE PROPERTY STRINGS CPU_ONLY)

# # Test for CUDA or OpenCL
# find_package(CUDA)
# if (CUDA_FOUND)
#   set_property(CACHE GPU_MODE PROPERTY STRINGS CUDA CPU_ONLY)
# endif ()
# # set_property(CACHE GPU_MODE PROPERTY STRINGS CUDA OPENCL CPU_ONLY)
B
Bikramjot Hanzra 已提交
118
set_property(CACHE GPU_MODE PROPERTY STRINGS CUDA)
G
gineshidalgo99 已提交
119

120 121 122
if (${GPU_MODE} MATCHES "CUDA")
  # OpenPose flags
  add_definitions(-DUSE_CUDA)
G
gineshidalgo99 已提交
123
  message(STATUS "Building with CUDA")
124 125 126
elseif (${GPU_MODE} MATCHES "CPU_ONLY")
  # OpenPose flag for Caffe
  add_definitions(-DCPU_ONLY)
G
gineshidalgo99 已提交
127 128 129 130 131 132 133 134
  message(STATUS "Building CPU Only")
endif ()

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

137 138 139 140 141
# Set/disable profiler
if (PROFILER_ENABLED)
  add_definitions(-DPROFILER_ENABLED)
endif (PROFILER_ENABLED)

B
Bikramjot Hanzra 已提交
142 143 144 145 146 147 148 149 150 151
# 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")
152 153

# Suboptions for OpenPose 3D Reconstruction demo
154
option(BUILD_MODULE_3D "Build OpenPose 3D Reconstruction module." OFF)
155

B
Bikramjot Hanzra 已提交
156 157 158 159 160 161 162 163 164 165
# Download the models
option(DOWNLOAD_COCO_MODEL "Download COCO model." ON)
option(DOWNLOAD_MPI_MODEL "Download MPI model." OFF)
option(DOWNLOAD_HAND_MODEL "Download hand model." ON)  
option(DOWNLOAD_FACE_MODEL "Download face model." ON)

# More options
option(BUILD_EXAMPLES "Build OpenPose examples." ON)
option(BUILD_DOCS "Build OpenPose documentation." OFF)

B
bikz05 已提交
166 167 168
# Build as shared library
option(BUILD_SHARED_LIBS "Build as shared lib" ON)

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

B
Bikramjot Hanzra 已提交
172

G
gineshidalgo99 已提交
173 174

### FIND REQUIRED PACKAGES
B
Bikramjot Hanzra 已提交
175
list(APPEND CMAKE_MODULE_PATH "${CMAKE_CURRENT_SOURCE_DIR}/cmake/Modules")
176

G
gineshidalgo99 已提交
177 178 179 180
if (${GPU_MODE} MATCHES "CUDA")
  find_package(CUDA)
endif ()

181 182 183 184 185 186
# Adding 3D
if (BUILD_MODULE_3D)
  # OpenPose flags
  add_definitions(-DBUILD_MODULE_3D)
endif (BUILD_MODULE_3D)

187
if (UNIX AND NOT APPLE)
G
gineshidalgo99 已提交
188 189 190 191
  if (${GPU_MODE} MATCHES "CUDA")
    include(cmake/Cuda.cmake)
    find_package(CuDNN)
  endif ()
192 193 194 195
  find_package(GFlags)
  find_package(Glog)
  find_package(OpenCV)

196 197 198 199 200 201 202 203 204 205 206 207
  # 3D
  if (BUILD_MODULE_3D)
    # GLUT
    find_package(GLUT REQUIRED)
    # Spinnaker
    find_package(Spinnaker)
    if (NOT SPINNAKER_FOUND)
      message(FATAL_ERROR "Spinnaker not found. Either turn off the `BUILD_MODULE_3D` option or specify the path of
        Spinnaker includes and libs.")
    endif (NOT SPINNAKER_FOUND)
  endif (BUILD_MODULE_3D)

G
gineshidalgo99 已提交
208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233
  # OpenMP
  if (${GPU_MODE} MATCHES "CPU_ONLY")
    find_package(OpenMP)
    if (OPENMP_FOUND)
      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 ()
  endif ()

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

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

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

  if (NOT GLOG_FOUND)
G
gineshidalgo99 已提交
236 237
    message(FATAL_ERROR "Glog not found. Install Glog from the command line using the command(s) -\
      sudo apt-get install libgoogle-glog-dev")
238 239 240
  endif (NOT GLOG_FOUND)

  if (NOT GFLAGS_FOUND)
G
gineshidalgo99 已提交
241
    message(FATAL_ERROR "GFlags not found. Install GFlags from the command line using the command(s) --\
242 243 244 245 246 247 248 249 250 251 252 253 254 255
      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)

endif (UNIX AND NOT APPLE)

include(cmake/Utils.cmake)

if (WIN32)

G
gineshidalgo99 已提交
256
  if ("${CMAKE_VERSION}" VERSION_GREATER 3.6.3)
257
    set_property(DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR} PROPERTY VS_STARTUP_PROJECT OpenPoseDemo)
G
gineshidalgo99 已提交
258
  endif ("${CMAKE_VERSION}" VERSION_GREATER 3.6.3)
259 260 261 262 263 264 265 266 267 268 269 270

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

  #TODO Add hashes
  download_zip("opencv_310.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} 1e5240a64b814b3c0b822f136be78ad7)
  download_zip("caffe3rdparty_2017_07_14.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} ec0f800c8fb337e33304f3375bd06a80)
G
gineshidalgo99 已提交
271
  download_zip("caffe_2018_01_18.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} 4b8e548cc7ea20abea472950dd5301bd)
272
  if (BUILD_MODULE_3D)
273
    download_zip("freeglut_2018_01_14.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} BB182187285E06880F0EDE3A39530091)
274
  endif (BUILD_MODULE_3D)
275 276 277 278 279 280 281 282 283 284
  message(STATUS "Windows dependencies downloaded.")

  find_library(OpenCV_LIBS opencv_world310 HINTS ${FIND_LIB_PREFIX}/opencv/x64/vc14/lib)
  find_library(GFLAGS_LIBRARY_RELEASE gflags HINTS ${FIND_LIB_PREFIX}/caffe3rdparty/lib)
  find_library(GFLAGS_LIBRARY_DEBUG gflagsd HINTS ${FIND_LIB_PREFIX}/caffe3rdparty/lib)
  find_library(GLOG_LIBRARY_RELEASE glog HINTS ${FIND_LIB_PREFIX}/caffe3rdparty/lib)
  find_library(GLOG_LIBRARY_DEBUG glogd HINTS ${FIND_LIB_PREFIX}/caffe3rdparty/lib)
  find_library(OpenCV_LIBS opencv_world310 HINTS ${FIND_LIB_PREFIX}/opencv/x64/vc14/lib)
  find_library(Caffe_LIB caffe HINTS ${FIND_LIB_PREFIX}/caffe/lib)
  find_library(Caffe_Proto_LIB caffeproto HINTS ${FIND_LIB_PREFIX}/caffe/lib)
285 286 287 288 289
  if (BUILD_MODULE_3D)
    find_library(SPINNAKER_LIB spinnaker_v140 HINTS ${FIND_LIB_PREFIX}/spinnaker/lib)
    find_library(GLUT_LIBRARY freeglut HINTS ${FIND_LIB_PREFIX}/freeglut/lib)
    message(STATUS "\${GLUT_LIBRARY} = ${GLUT_LIBRARY}")
  endif (BUILD_MODULE_3D)
290 291 292 293 294
  set(Caffe_LIBS ${Caffe_LIB};${Caffe_Proto_LIB})
  set(OpenCV_INCLUDE_DIRS "3rdparty/windows/opencv/include")
  set(Caffe_INCLUDE_DIRS "3rdparty/windows/caffe/include;3rdparty/windows/caffe/include2")
  set(Boost_INCLUDE_DIRS "3rdparty/windows/caffe3rdparty/include/boost-1_61")
  set(WINDOWS_INCLUDE_DIRS "3rdparty/windows/caffe3rdparty/include")
295
  if (BUILD_MODULE_3D)
296
    set(SPINNAKER_INCLUDE_DIRS "3rdparty/windows/spinnaker/include")
297 298
    set(GLUT_INCLUDE_DIRS "3rdparty/windows/freeglut/include")
  endif (BUILD_MODULE_3D)
299
endif (WIN32)
B
Bikramjot Hanzra 已提交
300 301


G
gineshidalgo99 已提交
302 303

### CAFFE
304
if (UNIX AND NOT APPLE)
B
Bikramjot Hanzra 已提交
305

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

308 309 310 311 312 313
    # 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 已提交
314

315
    # Check if caffe is installed in known paths
G
gineshidalgo99 已提交
316
    if (NOT Caffe_FOUND AND NOT BUILD_CAFFE)
317 318 319
      message(STATUS "Looking for caffe around in expected paths.")
      find_package(Caffe)
    endif (NOT Caffe_FOUND AND NOT BUILD_CAFFE)
G
Gines 已提交
320

321 322
    # Else build from scratch
    if (BUILD_CAFFE)
B
Bikramjot Hanzra 已提交
323

324 325
      # Download Caffe
      message(STATUS "Caffe will be downloaded from source now. NOTE: This process might take several minutes depending
G
gineshidalgo99 已提交
326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366
        on your internet connection.")

      # Check if pulled
      file(GLOB CAFFE_DIR_VALID ${CMAKE_SOURCE_DIR}/3rdparty/caffe/*)
      list(LENGTH CAFFE_DIR_VALID CAFFE_DIR_VALID_LENGTH)
      if (CAFFE_DIR_VALID_LENGTH EQUAL 0)
        execute_process(COMMAND git submodule update --init --recursive)
      else()
        message(STATUS "Caffe has already been downloaded.")
      endif ()

      # Build Process
      set(CAFFE_CPU_ONLY OFF)
      if (${GPU_MODE} MATCHES "CUDA")
        execute_process(COMMAND git checkout master WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
      elseif (${GPU_MODE} MATCHES "CPU_ONLY")
        if (USE_MKL)
          execute_process(COMMAND git checkout intel WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
          execute_process(COMMAND sh prepare_mkl.sh WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/3rdparty/caffe/external/mkl OUTPUT_VARIABLE rv)
          set( MLIST ${rv} )
          separate_arguments(MLIST)
          list(GET MLIST 0 MKL_PATH)
          file(GLOB MKL_SO
            "${MKL_PATH}lib/*"
            )
          file(COPY ${MKL_SO} DESTINATION ${CMAKE_BINARY_DIR}/caffe)

          set(MKL_LIBS
            "${CMAKE_BINARY_DIR}/caffe/libiomp5.so"
            "${CMAKE_BINARY_DIR}/caffe/libmklml_intel.so"
            "${CMAKE_BINARY_DIR}/caffe/lib/libmkldnn.so.0"
            "${CMAKE_BINARY_DIR}/caffe/lib/libmkldnn.so"
            "${CMAKE_BINARY_DIR}/caffe/lib/libmkldnn.so.0.12.0"
            )
        else ()
          execute_process(COMMAND git checkout master WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
        endif ()
        set(CAFFE_CPU_ONLY ON)
        set(USE_CUDNN OFF)
      endif ()

367 368 369 370 371
      # Build Caffe
      message(STATUS "Caffe will be built from source now.")
      include(ExternalProject)
      set(CAFFE_PREFIX caffe)
      set(CAFFE_URL ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
B
Bikramjot Hanzra 已提交
372

G
gineshidalgo99 已提交
373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393
      # One for Intel Branch and one for Master
      if (USE_MKL)
        ExternalProject_Add(openpose_caffe
          SOURCE_DIR ${CAFFE_URL}
          PREFIX ${CAFFE_PREFIX}
          CMAKE_ARGS -DCMAKE_INSTALL_PREFIX:PATH=<INSTALL_DIR>
          -DMKLDNN_INSTALL_DIR:PATH=<INSTALL_DIR>
          -DUSE_CUDNN=${USE_CUDNN}
          -DBUILD_python=OFF
          -DCPU_ONLY=${CAFFE_CPU_ONLY}
          -DUSE_MKL2017_AS_DEFAULT_ENGINE=${CAFFE_CPU_ONLY}
          -DCMAKE_BUILD_TYPE=Release
          -DBUILD_docs=OFF
          -DBUILD_python=OFF
          -DBUILD_python_layer=OFF
          -DUSE_LEVELDB=OFF
          -DUSE_LMDB=OFF
          -DUSE_OPENCV=OFF)
          # -DOpenCV_DIR=${OpenCV_DIR})
      else ()
        ExternalProject_Add(openpose_caffe
394 395
          SOURCE_DIR ${CAFFE_URL}
          PREFIX ${CAFFE_PREFIX}
G
gineshidalgo99 已提交
396
          CMAKE_ARGS -DCMAKE_INSTALL_PREFIX:PATH=<INSTALL_DIR>
G
gineshidalgo99 已提交
397 398 399 400 401 402 403 404 405 406 407
          -DUSE_CUDNN=${USE_CUDNN}
          -DCPU_ONLY=${CAFFE_CPU_ONLY}
          -DCMAKE_BUILD_TYPE=Release
          -DBUILD_docs=OFF
          -DBUILD_python=OFF
          -DBUILD_python_layer=OFF
          -DUSE_LEVELDB=OFF
          -DUSE_LMDB=OFF
          -DUSE_OPENCV=OFF)
          # -DOpenCV_DIR=${OpenCV_DIR})
      endif ()
B
Bikramjot Hanzra 已提交
408

409 410 411 412 413 414 415 416 417
      ExternalProject_Get_Property(openpose_caffe install_dir)

      set(Caffe_INCLUDE_DIRS ${CMAKE_BINARY_DIR}/caffe/include)
      # TODO -- Find a better soln.
      set(Caffe_LIBS ${CMAKE_BINARY_DIR}/caffe/lib/libcaffe.so)

    endif (BUILD_CAFFE)

    if (NOT Caffe_FOUND AND NOT BUILD_CAFFE)
G
gineshidalgo99 已提交
418
      message(FATAL_ERROR "Caffe not found. Either turn on the BUILD_CAFFE option or specify the path of Caffe includes
419
        and libs using -DCaffe_INCLUDE_DIRS and -DCaffe_LIBS.")
420 421 422 423 424
    endif (NOT Caffe_FOUND AND NOT BUILD_CAFFE)

  endif (${DL_FRAMEWORK} MATCHES "CAFFE")

endif (UNIX AND NOT APPLE)
B
Bikramjot Hanzra 已提交
425 426 427 428

### PROJECT INCLUDES
# Specify the include directories
include_directories(
429
  include
B
Bikramjot Hanzra 已提交
430 431 432 433
  ${GFLAGS_INCLUDE_DIR}
  ${GLOG_INCLUDE_DIR}
  ${Caffe_INCLUDE_DIRS}
  ${OpenCV_INCLUDE_DIRS})
G
gineshidalgo99 已提交
434 435 436 437
if (${GPU_MODE} MATCHES "CUDA")
  include_directories(
    ${CUDA_INCLUDE_DIRS})
endif ()
438 439 440 441 442
# 3D
if (BUILD_MODULE_3D)
  include_directories(${GLUT_INCLUDE_DIRS} ${SPINNAKER_INCLUDE_DIRS})
endif (BUILD_MODULE_3D)
# Windows includes
443 444 445 446 447
if (WIN32)
  include_directories(
    ${Boost_INCLUDE_DIRS}
    ${WINDOWS_INCLUDE_DIRS})
endif (WIN32)
B
Bikramjot Hanzra 已提交
448 449 450 451 452 453 454 455 456

### ADD SUBDIRECTORIES

add_subdirectory(src)
if (BUILD_EXAMPLES)
  add_subdirectory(examples)
endif (BUILD_EXAMPLES)


G
gineshidalgo99 已提交
457
### DOWNLOAD MODELS
B
Bikramjot Hanzra 已提交
458 459 460 461 462
# 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 已提交
463

464
download_model("body (COCO)" ${DOWNLOAD_COCO_MODEL} pose/coco/pose_iter_440000.caffemodel
G
gineshidalgo99 已提交
465
  5156d31f670511fce9b4e28b403f2939) # Body (COCO)
466
download_model("body (MPI)" ${DOWNLOAD_MPI_MODEL} pose/mpi/pose_iter_160000.caffemodel
G
gineshidalgo99 已提交
467
  2ca0990c7562bd7ae03f3f54afa96e00) # Body (MPI)
468
download_model("face" ${DOWNLOAD_FACE_MODEL} face/pose_iter_116000.caffemodel
G
gineshidalgo99 已提交
469
  e747180d728fa4e4418c465828384333) # Face
470
download_model("hand" ${DOWNLOAD_HAND_MODEL} hand/pose_iter_102000.caffemodel
G
gineshidalgo99 已提交
471
  a82cfc3fea7c62f159e11bd3674c1531) # Hand
B
Bikramjot Hanzra 已提交
472 473

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

475

G
gineshidalgo99 已提交
476
### GENERATE DOCUMENTATION
477 478 479 480 481 482 483 484 485 486
if (UNIX AND NOT APPLE)

  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 已提交
487 488 489 490
        COMMAND ${DOXYGEN_EXECUTABLE} ${DOXYGEN_FILE}
        WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/doc
        COMMENT "Generating API documentation with Doxygen"
        VERBATIM)
491
    else (DOXYGEN_FOUND)
492
      message(FATAL_ERROR "Doxygen need to be installed to generate the doxygen documentation.")
493 494 495 496 497
    endif (DOXYGEN_FOUND)
  endif (BUILD_DOCS)

endif (UNIX AND NOT APPLE)

B
bikz05 已提交
498

G
gineshidalgo99 已提交
499
### INSTALL
500 501 502 503 504 505 506 507 508 509 510
if (UNIX AND NOT APPLE)
  # 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(
G
gineshidalgo99 已提交
511 512
    ${CMAKE_SOURCE_DIR}/cmake/OpenPoseConfig.cmake.in
    ${CMAKE_BINARY_DIR}/cmake/OpenPoseConfig.cmake @ONLY)
513 514

  install(FILES ${CMAKE_BINARY_DIR}/cmake/OpenPoseConfig.cmake
G
gineshidalgo99 已提交
515
    DESTINATION lib/OpenPose)
516 517 518

  # Uninstall target
  configure_file(
G
gineshidalgo99 已提交
519 520 521
    "${CMAKE_SOURCE_DIR}/cmake/cmake_uninstall.cmake.in"
    "${CMAKE_BINARY_DIR}/cmake_uninstall.cmake"
    IMMEDIATE @ONLY)
522 523

  add_custom_target(uninstall
G
gineshidalgo99 已提交
524 525
    COMMAND ${CMAKE_COMMAND} -P ${CMAKE_CURRENT_BINARY_DIR}/cmake_uninstall.cmake)
endif (UNIX AND NOT APPLE)