cmake_minimum_required(VERSION 3.5)
|
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DGPU -DCUDNN -DOPENCV -g -std=c++11 -W -O3 -pthread")
|
set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -DGPU -DCUDNN -DOPENCV -g -std=c++11 -W -O3")
|
project(demo)
|
set(EXECUTABLE_OUTPUT_PATH ${PROJECT_SOURCE_DIR}/build)
|
|
set(CMAKE_CXX_STANDARD 11)
|
add_compile_options(-fPIC -g -Wall -Wshadow -Wno-sign-compare -w)
|
|
file(GLOB_RECURSE CURRENT_HEADERS
|
./*.h
|
./lib/*.h
|
./lib/core/*.h
|
./lib/tracker_tools/*.h
|
./lib/encoder_tools/*.h
|
./lib/detecter_tools/*.h
|
./lib/detecter_tools/extra/*.h
|
./lib/Munkres_assign/*.h
|
./lib/utils/*.h
|
./lib/detecter_tools/*.hpp
|
./lib/detecter_tools/*.cuh
|
./lib/detecter_tools/darknet/*.hpp
|
./lib/detecter_tools/darknet/*.cuh)
|
|
file(GLOB sources
|
./*.cpp
|
./lib/*.cpp
|
./lib/core/*.cpp
|
./lib/tracker_tools/*.cpp
|
./lib/encoder_tools/*.cpp
|
./lib/detecter_tools/*.cpp
|
./lib/Munkres_assign/*.cpp
|
./lib/utils/*.cpp
|
./lib/detecter_tools/*.cpp
|
./lib/detecter_tools/darknet/*.c
|
./lib/detecter_tools/darknet/*.cpp
|
./lib/detecter_tools/darknet/*.cu)
|
|
|
source_group("Include" FILES ${CURRENT_HEADERS})
|
source_group("Source" FILES ${sources})
|
|
|
#cudaS
|
find_package(CUDA REQUIRED)
|
|
message("-- CUDA version:$(CUDA_VERSION)")
|
|
if(CUDA_VERSION_MAJOR GREATER 9)
|
message("-- CUDA ${CUDA_VERSION_MAJOR} detected")
|
set(
|
CUDA_NVCC_FLAGS
|
${CUDA_NVCC_FLAGS};
|
-gencode arch=compute_61,code=sm_61
|
)
|
set(CUDA_NVCC_FLAGS_RELWITHDEBINFO "--device-debug;-lineinfo")
|
endif()
|
|
|
set(TENSORFLOW_DIR /home/disk1/s_opt/tensorflow/tensorflow-1.14.0)
|
|
link_directories(/usr/local/cuda-10.0/lib64)
|
#link_directories(/usr/local/lib)
|
|
SET(LIBS
|
opencv_core
|
opencv_highgui
|
opencv_imgcodecs
|
opencv_imgproc
|
opencv_videoio
|
opencv_video
|
tensorflow_cc
|
tensorflow_framework
|
cudart
|
cublas
|
stdc++fs
|
curand
|
cuda
|
cudnn
|
)
|
|
include_directories(
|
${CMAKE_CURRENT_SOURCE_DIR}
|
/home/disk1/s_opt/tensorflow/tensorflow-1.14.0
|
/home/disk1/s_opt/tensorflow/tensorflow-1.14.0/bazel-genfiles
|
/home/disk1/s_opt/tensorflow/tensorflow-1.14.0/bazel-bin/tensorflow
|
/home/disk1/s_opt/tensorflow/tensorflow-1.14.0/tensorflow/contrib/makefile/downloads/nsync/public
|
/home/disk1/s_opt/tensorflow/tensorflow-1.14.0/tensorflow/contrib/makefile/downloads/eigen
|
/home/disk1/s_opt/tensorflow/tensorflow-1.14.0/tensorflow/contrib/makefile/downloads/absl
|
/home/disk1/s_opt/tensorflow/tensorflow-1.14.0/tensorflow/contrib/makefile/downloads/protobuf/src
|
)
|
include_directories(/home/disk1/s_opt/opencv/opencv3.4.8/include) # 使用了系统的opencv，所以就不用这个了
|
include_directories(/usr/local/cuda/include)
|
include_directories(".")
|
include_directories("./lib")
|
include_directories("./lib/core")
|
include_directories("./lib/detecter_tools")
|
include_directories("./lib/encoder_tools")
|
include_directories("./lib/Munkres_assign")
|
include_directories("./lib/tracker_tools")
|
include_directories("./lib/utils")
|
include_directories("./lib/Munkres_assign/munkres")
|
link_directories(${TENSORFLOW_DIR}/tensorflow/contrib/makefile/downloads/nsync/builds/default.linux.c++11)
|
link_directories(${TENSORFLOW_DIR}/bazel-bin/tensorflow)
|
#link_directories(/usr/local/lib)
|
link_directories(/usr/local/cuda/lib64) #
|
link_directories(/home/disk1/s_opt/opencv/opencv3.4.8/lib)
|
#link_directories("/usr/lib/x86_64-linux-gnu")
|
|
|
cuda_add_executable(${PROJECT_NAME}
|
SDK_main.cpp
|
${sources}
|
${CURRENT_HEADERS}
|
)
|
target_link_libraries(${PROJECT_NAME}
|
${LIBS}
|
)
|
