cmake_minimum_required(VERSION 3.5)
|
|
#SET(CMAKE_CXX_FLAGS_DEBUG "$ENV{CXXFLAGS} -fvisibility=hidden -DEIGEN_MAX_ALIGN_BYTES=32 -DEIGEN_DONT_VECTORIZE -march=native -O0 -Wall -g2 -ggdb")
|
#set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fvisibility=hidden -DEIGEN_MAX_ALIGN_BYTES=32 -DEIGEN_DONT_VECTORIZE -fPIC -march=native -DGPU -DCUDNN -DOPENCV -g -std=c++14 -W -O3 -pthread -Wl,--no-as-needed -DS_DEBUG")
|
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fvisibility=hidden -DEIGEN_MAX_ALIGN_BYTES=32 -DEIGEN_DONT_VECTORIZE -fPIC -lm -march=native -DGPU -DCUDNN -DOPENCV -g -std=c++11 -W -O3 -pthread -Wl,--no-as-needed")
|
set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -fvisibility=hidden -DGPU -DCUDNN -DOPENCV -g -lm -std=c++11 -fPIC -W -O3 -Wl,--no-as-needed") # 编译C脚本时调用GPU环境信息
|
#set(CMAKE_C_FLAGS "-fvisibility=hidden")
|
#set(CMAKE_CXX_FLAGS "-fvisibility=hidden")
|
|
#项目名称/工程名
|
project(yolo)
|
set(EXECUTABLE_OUTPUT_PATH ${PROJECT_SOURCE_DIR}/build)
|
#设置c++编译器
|
|
set(CMAKE_CXX_STANDARD 11)
|
#set(CMAKE_BUILD_TYPE debug)
|
add_compile_options(-fPIC -ldl -lm -g -Wall -Wshadow -Wno-sign-compare -w -pthread)
|
#add_definitions(-DDEBUG_ERR -DDEBUG_INFO)
|
#add_definitions(-DDEBUG)
|
#add_definitions(-D_GLIBCXX_USE_CXX11_ABI=0)
|
|
|
file(GLOB_RECURSE CURRENT_HEADERS
|
./*.h
|
./lib/*.h
|
./lib/core/*.h
|
./lib/detecter_tools/*.h
|
./lib/detecter_tools/extra/*.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/detecter_tools/*.cpp
|
./lib/utils/*.cpp
|
./lib/detecter_tools/*.cpp
|
./lib/detecter_tools/darknet/*.c
|
./lib/detecter_tools/darknet/*.cpp
|
./lib/detecter_tools/darknet/*.cu)
|
|
#file(GLOB protobuf_files
|
# ./lib/additional/protomsg.proto
|
#)
|
#foreach(s range ${sources})
|
#message(${s})
|
#endforeach()
|
|
source_group("Include" FILES ${CURRENT_HEADERS})
|
source_group("Source" FILES ${sources})
|
|
|
#cudaS
|
find_package(CUDA REQUIRED)
|
|
#message("-- CUDA version:$(CUDA_VERSION)")
|
|
# 设置环境的模型查找，如果CUDA版本大于9,则执行CUDA环境的模型查找
|
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 -std=c++11# 不同GPU有不同的算力指数，可查看算力表
|
)
|
set(CUDA_NVCC_FLAGS_RELWITHDEBINFO "--device-debug;-lineinfo")
|
#find_package(OpenCV REQUIRED) # 查找系统的默认opencv环境
|
#message(${OpenCV_LIBS})
|
endif()
|
|
|
set(OPENCV_DIR /home/disk1/s_opt/01_opencv/opencv2.4.13)
|
|
link_directories(/usr/local/cuda-10.0/lib64)
|
#link_directories(/usr/local/lib)
|
|
SET(LIBS
|
cudart
|
gomp
|
nvToolsExt
|
opencv_core
|
opencv_highgui
|
#opencv_imgcodecs
|
opencv_imgproc
|
#opencv_videoio
|
opencv_video
|
opencv_calib3d
|
cublas
|
stdc++fs #设置-lstdc++fs 新链接器选项，防止std等gcc库调用不到
|
curand
|
cuda
|
cudnn
|
jsoncpp
|
)
|
|
# 将源码目录保存到变量中
|
#aux_source_directory(./src DIR_SRCS) # 搜索当前目录下的所有.cpp文件
|
#add_library(demo ${SRC_LIST}) #明确指定包含哪些源文件
|
|
|
#设置包含的目录，项目中的include路径，换成自己的路径即可
|
include_directories(/usr/local/cuda-10.0/include)
|
include_directories(".")
|
include_directories("./lib")
|
include_directories("./lib/core")
|
include_directories("./lib/detecter_tools")
|
include_directories("./lib/utils")
|
#include_directories("/usr/include/x86_64-linux-gnu/")
|
#include_directories("/usr/include/json/")
|
include_directories("/usr/include")
|
include_directories("/usr/local/include/")
|
include_directories(${OPENCV_DIR}/include)
|
|
#设置链接库搜索目录，项目中lib路径
|
link_directories(${OPENCV_DIR}/lib)
|
link_directories(/usr/local/cuda-10.0/lib64) #动态链接库目录/bazel-bin/tensorflow
|
link_directories(/usr/local/lib) #动态链接库目录/bazel-bin/tensorflow
|
link_directories(/usr/lib/x86_64-linux-gnu/)
|
link_directories(/home/basic/anaconda3/lib/)
|
|
|
#添加要编译的可执行文件
|
#cuda_add_executable(${PROJECT_NAME}
|
|
cuda_add_executable(${PROJECT_NAME} SDK_main.cpp ${sources} ${CURRENT_HEADERS})
|
#cuda_add_library(${PROJECT_NAME} SHARED ${sources} ${CURRENT_HEADERS})
|
|
#add_executable(demo ${DIR_SRCS}) ## 生成可执行文件
|
|
|
#设置 target 需要链接的库
|
#添加可执行文件所需要的库,连接libtensorflow_cc.so和libtensorflow_framework库，链接动态链接库
|
target_link_libraries(${PROJECT_NAME} ${LIBS})
|
