Yolov5 的 C++ 实现，用于检测 Jetson Xavier nx 和 Jetson nano 中野外的火灾或烟雾

cmake_minimum_required(VERSION 2.6) project(yolofire) list(APPEND CUDA_NVCC_FLAGS "-std=c++11") set(CMAKE_CXX_FLAGS "-std=c++0x") find_package(OpenCV REQUIRED) add_definitions(-std=c++11) add_definitions(-DAPI_EXPORTS) option(CUDA_USE_STATIC_CUDA_RUNTIME OFF) set(CMAKE_CXX_STANDARD 11) set(CMAKE_BUILD_TYPE Release) find_package(CUDA REQUIRED) set(CUDA_NVCC_PLAGS ${CUDA_NVCC_PLAGS};-std=c++11;-g;-G;-gencode;arch=compute_53;code=sm_53) find_package(OpenCV REQUIRED core imgproc highgui) if(WIN32) enable_language(CUDA) endif(WIN32) # include and link dirs of cuda and tensorrt, you need adapt them if yours are different # cuda include_directories(/usr/local/cuda/include) link_directories(/usr/local/cuda/lib64) # tensorrt include_directories(/usr/include/aarch64-linux-gnu/) link_directories(/usr/lib/aarch64-linux-gnu/) include_directories( ${CUDA_INCLUDE_DIRS} ${OpenCV_INCLUDE_DIRS} ) # ===== yolo ===== include_directories(${PROJECT_SOURCE_DIR}/yolo/include) aux_source_directory(${PROJECT_SOURCE_DIR}/yolo/src YOLO_SRC_DIR) set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11 -Wall -Ofast -Wfatal-errors -D_MWAITXINTRIN_H_INCLUDED") cuda_add_library(yolov5_trt SHARED ${PROJECT_SOURCE_DIR}/yolo/include/yololayer.cu ${PROJECT_SOURCE_DIR}/yolo/src/yolov5_lib.cpp) target_link_libraries(yolov5_trt nvinfer cudart) # ===== main ===== aux_source_directory(${PROJECT_SOURCE_DIR}/src M_SRC_DIR) include_directories(${PROJECT_SOURCE_DIR}/include) add_executable(yolofire ${M_SRC_DIR}) target_link_libraries(yolofire nvinfer cudart yolov5_trt ${OpenCV_LIBS}) if(UNIX) add_definitions(-O2 -pthread) endif(UNIX) set(CMAKE_CXX_FLAGS "-std=c++0x")