From e2e5f3d59e24975dd99ec5aa8618760ae1214a16 Mon Sep 17 00:00:00 2001 From: Scheaven <xuepengqiang> Date: 星期二, 14 九月 2021 10:58:46 +0800 Subject: [PATCH] cmake bug --- readme.txt | 4 ++++ CMakeLists.txt | 10 +++++++--- 2 files changed, 11 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( diff --git a/readme.txt b/readme.txt new file mode 100644 index 0000000..0f0a17d --- /dev/null +++ b/readme.txt @@ -0,0 +1,4 @@ +config.json 鏄厤缃俊鎭殑鏂囦欢 +model_path 鏄浆鍖栧彿鐨則rt妯″瀷璺緞 +濡傛灉娌℃湁杞寲濂界殑妯″瀷鏂囦欢锛岄渶瑕侀厤缃甿odel_cfg锛歝fg鏂囦欢鍜宮odel_wts锛歸eights鏉冮噸鏂囦欢銆傚鏋渕odel_path瀛樺湪鍒欎笉闇�瑕� +"type"瀛楁濡傛灉鏄�1锛屽垯涓烘甯哥増鏈紝濡傛灉涓�2锛屽垯鏄痶iny鐗堟湰 -- Gitblit v1.8.0