一个使用Tensorrt进行推理的应用库

project(PerceptionDataMasking C CXX) include_directories( Inc ${WORK_BASE}/Include ${WORK_BASE}/Common/Inc ${WORK_BASE}/NvmAdaptor/Inc ) include_directories(Plugin/common/Inc Plugin/Src) aux_source_directory(Src BBOX_SOURCE_FILES) aux_source_directory(Plugin/Src BBOX_SOURCE_FILES) aux_source_directory(Plugin/common/Src BBOX_SOURCE_FILES) include_directories(${WORK_BASE}/Include) include_directories(${WORK_BASE}/3rdParty/qnx-aarch64/jemalloc/include) set(OPENCV_LIBS ${OPENCV_LIBS} opencv_imgcodecs opencv_ml opencv_core opencv_imgproc opencv_video) link_directories( ${CUDNN_DIR}/lib/aarch64-unknown-nto-qnx ${CUDNN_DIR}/lib ${QNX_TARGET}/aarch64le/lib ${OPENCV_DIR}/lib ) link_directories( ${CUDA_DIR}/nvvm/lib64 ${DRIVE_PDK}/lib-target ${WORK_BASE}/3rdParty/qnx-aarch64/protobuf-3.11.4/libs ) link_directories(${DRIVE_PDK}/nvidia-bsp/aarch64le/usr/lib) set(OPENCV_LIBS ${OPENCV_LIBS} avcodec avformat avutil avdevice avfilter swscale avresample postproc swresample socket nvtime2 nvscisync) set(CUDA_CUDNN_LIBS nvdtcommon nvtegrahv nvscibuf nvos_s3_safety nvos nvdla_runtime nvrm_mem nvrm_host1x nvsciipc nvrm_chip nvivc nvsocsys nvdc) set(CUDA_CUDNN_LIBS ${CUDA_CUDNN_LIBS} nvparser nvrm_gpu nvscicommon nvrm_surface nvrm_stream nvmedia_tensor nvrm_sync nvdla_compiler icui18n nvmedia_ijpe_sci nvmedia_sci_overlay) set(CUDA_CUDNN_LIBS ${CUDA_CUDNN_LIBS} nvinfer_plugin nvparsers cuda cublas nvonnxparser cudnn curand cublasLt nvmedia_dla nvmedia_core nvinfer) set(CUDA_CUDNN_LIBS ${CUDA_CUDNN_LIBS} nvmedia2d nvvic nvvideo cuda nvscistream nvscibuf nvscisync nvscievent nvsciipc nvscicommon cudart) #------------------------------------------------------------------------------- # Final target #------------------------------------------------------------------------------- set(CMAKE_INSTALL_RPATH_USE_LINK_PATH TRUE) set(PERCEPTION_FEATURES "-Wl,-rpath-link,${DRIVE_PDK}/lib-target") set(PERCEPTION_FEATURES ${PERCEPTION_FEATURES} -Wl,--whole-archive -Wl,--no-whole-archive) #------------------------------------------------------------------------------- # Final target #------------------------------------------------------------------------------- add_library(${PROJECT_NAME} SHARED ${BBOX_SOURCE_FILES}) target_link_libraries(${PROJECT_NAME} ${PERCEPTION_FEATURES} ${CUDA_CUDNN_LIBS} ${OPENCV_LIBS} ${EXTRA_LIBS} )