| 123456789101112131415161718192021222324 |
- cmake_minimum_required(VERSION 3.0)
-
- project(testcv)
-
- set(OpenCV_DIR "~/lib/opencv4.10.0_install/lib/cmake/opencv4") #对应上面刚查的目录
- include_directories(/home/cl/package/TensorRT-8.6.1.6/include)
- link_directories(/home/cl/package/TensorRT-8.6.1.6/lib)
- find_package(CUDA REQUIRED)
- include_directories(/usr/local/cuda-12.1/include)
- # link_directories(/usr/local/cuda-12.1/lib64)
-
- find_package(OpenCV 4.10 REQUIRED)
- message(STATUS "> include dir: ${OpenCV_INCLUDE_DIRS}")
- include_directories(${OpenCV_INCLUDE_DIRS})
- add_executable(yolo-crowd yolov5_trt.cpp)
- message("find ${OpenCV_VERSION}")
- target_link_libraries(yolo-crowd ${OpenCV_LIBS})
- target_link_libraries(yolo-crowd ${CUDA_LIBRARIES})
- target_link_libraries(yolo-crowd nvinfer)
|
