From e2e5f3d59e24975dd99ec5aa8618760ae1214a16 Mon Sep 17 00:00:00 2001
From: Scheaven <xuepengqiang>
Date: 星期二, 14 九月 2021 10:58:46 +0800
Subject: [PATCH] cmake bug
---
CMakeLists.txt | 10 +++++++---
1 files changed, 7 insertions(+), 3 deletions(-)
diff --git a/CMakeLists.txt b/CMakeLists.txt
index 72caba6..a2fc127 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -7,10 +7,13 @@
set(CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} -Wl,-rpath -Wl,$ORIGIN")
set(CUDA_Path /usr/local/cuda)
-find_package(OpenCV REQUIRED PATHS "/data/disk2/opt/01_opencv/opencv4.5.2")
+#find_package(OpenCV REQUIRED PATHS "/data/disk2/opt/01_opencv/opencv4.5.2")
+set(OPENCV_DIR /data/disk2/opt/01_opencv/opencv4.5.2)
set(TensorRT_DIR /data/disk2/opt/05_tensorRT/TensorRT-7.2.3.4)
set(LOG_DIR /data/disk2/opt/log4cplus)
+include_directories(${OPENCV_DIR}/include/opencv4)
+link_directories(${OPENCV_DIR}/lib)
include_directories(${CUDA_Path}/include)
link_directories(${CUDA_Path}/lib64)
include_directories(${TensorRT_DIR}/include)
@@ -20,7 +23,8 @@
include_directories(${LOG_DIR}/include)
file(GLOB_RECURSE sources src/* src/core/* src/utils/* src/detecter_tools/* ${TensorRT_DIR}/samples/common/*.h ${TensorRT_DIR}/samples/common/logger.cpp)
-set(LIBS nvinfer nvparsers nvonnxparser nvinfer_plugin stdc++fs jsoncpp log4cplus)
+set(LIBS nvinfer nvparsers nvonnxparser nvinfer_plugin stdc++fs jsoncpp log4cplus opencv_core opencv_highgui opencv_imgcodecs
+ opencv_imgproc opencv_videoio opencv_video opencv_calib3d opencv_features2d opencv_flann opencv_dnn)
if(NOT DEFINED CMAKE_CUDA_ARCHITECTURES)
set(CMAKE_CUDA_ARCHITECTURES 61 75 72 80)
@@ -28,7 +32,7 @@
#
-
+find_package(CUDA REQUIRED)
if(CUDA_VERSION_MAJOR GREATER 9)
message("-- CUDA ${CUDA_VERSION_MAJOR} detected")
set(
--
Gitblit v1.8.0