CMakeLists.txt 1.3 KB
Newer Older
G
gineshidalgo99 已提交
1 2
set(CMAKE_CXX_SOURCE_FILE_EXTENSIONS C;M;c++;cc;cpp;cxx;mm;CPP;cl)
set(SOURCES_OP_NET
G
gineshidalgo99 已提交
3 4
    bodyPartConnectorBase.cpp
    bodyPartConnectorBase.cu
R
Raaj 已提交
5
    bodyPartConnectorBaseCL.cpp
G
gineshidalgo99 已提交
6
    bodyPartConnectorCaffe.cpp
G
gineshidalgo99 已提交
7 8 9 10
    maximumBase.cpp
    maximumBase.cu
    maximumCaffe.cpp
    netCaffe.cpp
G
gineshidalgo99 已提交
11
    netOpenCv.cpp
G
gineshidalgo99 已提交
12 13 14 15 16 17 18 19 20 21 22 23 24 25
    nmsBase.cpp
    nmsBase.cu
    nmsBaseCL.cpp
    nmsCaffe.cpp
    resizeAndMergeBase.cpp
    resizeAndMergeBase.cu
    resizeAndMergeBaseCL.cpp
    resizeAndMergeCaffe.cpp)

include(${CMAKE_SOURCE_DIR}/cmake/Utils.cmake)
prepend(SOURCES_OP_NET_WITH_CP ${CMAKE_CURRENT_SOURCE_DIR} ${SOURCES_OP_NET})
set(SOURCES_OP_NET_WITH_CP ${SOURCES_OP_NET_WITH_CP} PARENT_SCOPE)
set(SOURCES_OPENPOSE ${SOURCES_OPENPOSE} ${SOURCES_OP_NET_WITH_CP} PARENT_SCOPE)

26
if (UNIX OR APPLE)
G
gineshidalgo99 已提交
27 28
  if (${GPU_MODE} MATCHES "CUDA")
    cuda_add_library(openpose_net ${SOURCES_OP_NET})
29
  else ()
G
gineshidalgo99 已提交
30 31 32 33 34
    add_library(openpose_net ${SOURCES_OP_NET})
  endif ()

  add_library(caffe SHARED IMPORTED)
  set_property(TARGET caffe PROPERTY IMPORTED_LOCATION ${Caffe_LIBS}) 
35
  target_link_libraries(openpose_net caffe ${MKL_LIBS} openpose_core)
G
gineshidalgo99 已提交
36 37

  if (BUILD_CAFFE)
38
    add_dependencies(openpose_net openpose)
G
gineshidalgo99 已提交
39 40 41 42 43 44 45
  endif (BUILD_CAFFE)

  install(TARGETS openpose_net
      EXPORT OpenPose
      RUNTIME DESTINATION bin
      LIBRARY DESTINATION lib
      ARCHIVE DESTINATION lib/openpose)
46
endif (UNIX OR APPLE)