-
Notifications
You must be signed in to change notification settings - Fork 104
/
CMakeLists_deepsort-tensorrt_win10.txt
52 lines (42 loc) · 1.48 KB
/
CMakeLists_deepsort-tensorrt_win10.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
cmake_minimum_required(VERSION 2.6)
project(deepsort)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11")
set(CMAKE_BUILD_TYPE Release)
set(CUDA_BIN_PATH C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v11.1) #1.修改为自己的cuda路径
set(TRT_DIR "D:\\lbq\\TensorRT-7.2.3.4") #2.修改为自己的tensorRT路径
set(TRT_INCLUDE_DIRS ${TRT_DIR}\\include) #3.修改为自己的tensorRT头文件路径
set(TRT_LIB_DIRS ${TRT_DIR}\\lib) #4.修改为自己的tensorRT库文件路径
set(Eigen3_PATH D:\\lbq\\eigen) #5.修改为自己的Eigen路径
option(CUDA_USE_STATIC_CUDA_RUNTIME OFF)
find_package(OpenCV REQUIRED)
find_package(CUDA REQUIRED)
enable_language(CUDA)
include_directories(
${CUDA_INCLUDE_DIRS}
${OpenCV_INCLUDE_DIRS}
${TRT_INCLUDE_DIRS}
${Eigen3_PATH}
${PROJECT_SOURCE_DIR}/include
)
# tensorRT
link_directories(${TRT_LIB_DIRS})
link_directories(${OpenCV_LIB_DIRS})
aux_source_directory(${PROJECT_SOURCE_DIR}/src SRC_DIR)
# ===== deepsort =====
add_library(deepsort STATIC ${SRC_DIR})
target_link_libraries(deepsort
${CUDA_LIBRARIES} ${OpenCV_LIBS}
cudart nvinfer nvonnxparser
)
# ===== onnx2engine =====
add_executable(onnx2engine ${PROJECT_SOURCE_DIR}/onnx2engine.cpp)
target_link_libraries(onnx2engine
${CUDA_LIBRARIES}
cudart nvinfer nvonnxparser deepsort
)
# ===== demo =====
add_executable(demo ${PROJECT_SOURCE_DIR}/demo.cpp)
target_link_libraries(demo
${CUDA_LIBRARIES} ${OpenCV_LIBS}
cudart nvinfer nvonnxparser deepsort
)