onnx2tensorrt

cmake_minimum_required(VERSION 3.8) project(onnx2trt) if(CMAKE_COMPILER_IS_GNUCXX OR CMAKE_CXX_COMPILER_ID MATCHES "Clang") add_compile_options(-Wall -Wextra -Wpedantic) endif() # find dependencies find_package(ament_cmake REQUIRED) find_package(OpenCV REQUIRED COMPONENTS core highgui imgproc dnn) find_package(CUDA REQUIRED) # uncomment the following section in order to fill in # further dependencies manually. # find_package(<dependency> REQUIRED) # 包含目录配置 -------------------------------------------- include_directories( ${CUDA_INCLUDE_DIRS} ${OpenCV_INCLUDE_DIRS} /usr/include/x86_64-linux-gnu/ ) link_directories( /usr/lib/x86_64-linux-gnu/ /usr/local/cuda/lib64 ) # 编译目标配置 -------------------------------------------- add_executable(onnx2trt src/calibrator.cpp src/onnx2trt.cpp src/main.cpp ) target_link_libraries(onnx2trt ${OpenCV_LIBS} ${TensorRT_LIBRARIES} # 链接TensorRT核心库 ${CUDA_LIBRARIES} nvinfer nvinfer_plugin nvonnxparser # 显式链接TensorRT组件 cudart cublas cudnn # 明确链接CUDA库 ) install(TARGETS onnx2trt DESTINATION lib/${PROJECT_NAME} ) if(BUILD_TESTING) find_package(ament_lint_auto REQUIRED) # the following line skips the linter which checks for copyrights # comment the line when a copyright and license is added to all source files set(ament_cmake_copyright_FOUND TRUE) # the following line skips cpplint (only works in a git repo) # comment the line when this package is in a git repo and when # a copyright and license is added to all source files set(ament_cmake_cpplint_FOUND TRUE) ament_lint_auto_find_test_dependencies() endif() ament_package()