cmake_minimum_required(VERSION 2.8)
|
|
set(BIN wface)
|
project (${BIN})
|
|
##################################################################################
|
|
if(NOT CMAKE_BUILD_TYPE)
|
set(CMAKE_BUILD_TYPE "Release")
|
endif()
|
|
set(CMAKE_CXX_FLAGS "-fvisibility=default -fPIC -Wl,-Bsymbolic ${CMAKE_CXX_FLAGS}")
|
set(CMAKE_C_FLAGS "-fvisibility=default -fPIC -Wl,-Bsymbolic ${CMAKE_C_FLAGS}")
|
|
set(CMAKE_CXX_FLAGS_DEBUG "-w -g -O0 -std=c++11 ${CMAKE_CXX_FLAGS}")
|
set(CMAKE_C_FLAGS_DEBUG "-w -g -O0 ${CMAKE_C_FLAGS}")
|
|
set(CMAKE_CXX_FLAGS_RELEASE "-w -O3 -std=c++11 ${CMAKE_CXX_FLAGS}")
|
set(CMAKE_C_FLAGS_RELEASE "-w -O3 ${CMAKE_C_FLAGS}")
|
|
######################################################################################
|
|
include_directories(
|
${CMAKE_SOURCE_DIR}/sdk/include
|
${CMAKE_SOURCE_DIR}/csrc
|
)
|
|
set(CMAKE_LIBRARY_PATH ${CMAKE_LIBRARY_PATH}
|
/lib
|
/usr/lib
|
/usr/lib64
|
/usr/local/lib
|
/usr/local/lib64
|
/usr/lib/x86_64-linux-gnu
|
)
|
|
#############################################################################
|
|
set(cuda /usr/local/cuda-10.0)
|
|
set(CUDA_TOOLKIT_ROOT_DIR ${cuda})
|
include_directories(${cuda}/include)
|
find_package(CUDA QUIET REQUIRED)
|
link_directories(${cuda}/lib64)
|
set(LINK_LIB -lcudart -lcublas -lcurand)
|
|
############################################################################
|
|
file(GLOB_RECURSE SDK_LIST "csrc/*.cpp")
|
set (SRC_LIST
|
${SDK_LIST}
|
${CMAKE_SOURCE_DIR}/cface.cpp
|
)
|
|
link_directories(${CMAKE_SOURCE_DIR}/sdk/lib)
|
list(APPEND LINK_LIB -lTHFaceImage -lTHFeature -lTHFaceProperty -lTHFaceTracking -lTHFaceMask)
|
|
FOREACH(src ${SRC_LIST})
|
MESSAGE( ${src} )
|
ENDFOREACH()
|
|
FOREACH(src ${LINK_LIB})
|
MESSAGE( ${src} )
|
ENDFOREACH()
|
|
add_library(${BIN} SHARED ${SRC_LIST})
|
add_custom_command(
|
TARGET ${BIN}
|
POST_BUILD
|
COMMAND cp ${CMAKE_SOURCE_DIR}/build/lib${BIN}.so ${CMAKE_SOURCE_DIR}/../libs/
|
)
|
target_link_libraries(${BIN} ${LINK_LIB} rt pthread dl)
|
