| | |
| | | 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) |
| | |
| | | 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) |
| | |
| | | |
| | | |
| | | # |
| | | |
| | | find_package(CUDA REQUIRED) |
| | | if(CUDA_VERSION_MAJOR GREATER 9) |
| | | message("-- CUDA ${CUDA_VERSION_MAJOR} detected") |
| | | set( |