派生自 Algorithm/baseDetector

Scheaven
2021-09-14 e2e5f3d59e24975dd99ec5aa8618760ae1214a16
cmake bug
1个文件已添加
1个文件已修改
14 ■■■■ 已修改文件
CMakeLists.txt 10 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
readme.txt 4 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
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(
readme.txt
New file
@@ -0,0 +1,4 @@
config.json 是配置信息的文件
model_path 是转化号的trt模型路径
如果没有转化好的模型文件,需要配置model_cfg:cfg文件和model_wts:weights权重文件。如果model_path存在则不需要
"type"字段如果是1,则为正常版本,如果为2,则是tiny版本