├── .gitattributes ├── .gitignore ├── CMakeLists.txt ├── LICENSE ├── README.md ├── README_onnx_plngin.md ├── images ├── coco_1.jpg └── render.jpg ├── main.cpp ├── run_yolov5.sh ├── src ├── application │ ├── CMakeLists.txt │ └── yolov5 │ │ ├── yolo.cpp │ │ └── yolo.h ├── module │ ├── CMakeLists.txt │ ├── builder │ │ ├── trt_builder.cpp │ │ └── trt_builder.h │ ├── common │ │ ├── cuda_tools.cpp │ │ ├── cuda_tools.h │ │ ├── ilogger.cpp │ │ ├── ilogger.h │ │ └── utils.h │ ├── core │ │ ├── async_infer.h │ │ ├── monopoly_allocator.h │ │ ├── trt_tensor.cpp │ │ └── trt_tensor.h │ └── infer │ │ ├── trt_infer.cpp │ │ └── trt_infer.h └── onnxplugin │ ├── CMakeLists.txt │ ├── include │ ├── SiLUPlugin.h │ ├── checkMacrosPlugin.h │ ├── kernel.h │ └── plugin.h │ └── src │ ├── SiLU.cu │ └── SiLUPlugin.cpp └── weights ├── yolov5n.engine ├── yolov5n.onnx ├── yolov5n.plugin.engine └── yolov5n.plugin.onnx /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/.gitattributes -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- 1 | bin 2 | build 3 | lib 4 | .vscode 5 | *.so 6 | model/* 7 | -------------------------------------------------------------------------------- /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/README.md -------------------------------------------------------------------------------- /README_onnx_plngin.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/README_onnx_plngin.md -------------------------------------------------------------------------------- /images/coco_1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/images/coco_1.jpg -------------------------------------------------------------------------------- /images/render.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/images/render.jpg -------------------------------------------------------------------------------- /main.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/main.cpp -------------------------------------------------------------------------------- /run_yolov5.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/run_yolov5.sh -------------------------------------------------------------------------------- /src/application/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/src/application/CMakeLists.txt -------------------------------------------------------------------------------- /src/application/yolov5/yolo.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/src/application/yolov5/yolo.cpp -------------------------------------------------------------------------------- /src/application/yolov5/yolo.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/src/application/yolov5/yolo.h -------------------------------------------------------------------------------- /src/module/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/src/module/CMakeLists.txt -------------------------------------------------------------------------------- /src/module/builder/trt_builder.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/src/module/builder/trt_builder.cpp -------------------------------------------------------------------------------- /src/module/builder/trt_builder.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/src/module/builder/trt_builder.h -------------------------------------------------------------------------------- /src/module/common/cuda_tools.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/src/module/common/cuda_tools.cpp -------------------------------------------------------------------------------- /src/module/common/cuda_tools.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/src/module/common/cuda_tools.h -------------------------------------------------------------------------------- /src/module/common/ilogger.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/src/module/common/ilogger.cpp -------------------------------------------------------------------------------- /src/module/common/ilogger.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/src/module/common/ilogger.h -------------------------------------------------------------------------------- /src/module/common/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/src/module/common/utils.h -------------------------------------------------------------------------------- /src/module/core/async_infer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/src/module/core/async_infer.h -------------------------------------------------------------------------------- /src/module/core/monopoly_allocator.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/src/module/core/monopoly_allocator.h -------------------------------------------------------------------------------- /src/module/core/trt_tensor.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/src/module/core/trt_tensor.cpp -------------------------------------------------------------------------------- /src/module/core/trt_tensor.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/src/module/core/trt_tensor.h -------------------------------------------------------------------------------- /src/module/infer/trt_infer.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/src/module/infer/trt_infer.cpp -------------------------------------------------------------------------------- /src/module/infer/trt_infer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/src/module/infer/trt_infer.h -------------------------------------------------------------------------------- /src/onnxplugin/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/src/onnxplugin/CMakeLists.txt -------------------------------------------------------------------------------- /src/onnxplugin/include/SiLUPlugin.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/src/onnxplugin/include/SiLUPlugin.h -------------------------------------------------------------------------------- /src/onnxplugin/include/checkMacrosPlugin.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/src/onnxplugin/include/checkMacrosPlugin.h -------------------------------------------------------------------------------- /src/onnxplugin/include/kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/src/onnxplugin/include/kernel.h -------------------------------------------------------------------------------- /src/onnxplugin/include/plugin.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/src/onnxplugin/include/plugin.h -------------------------------------------------------------------------------- /src/onnxplugin/src/SiLU.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/src/onnxplugin/src/SiLU.cu -------------------------------------------------------------------------------- /src/onnxplugin/src/SiLUPlugin.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/src/onnxplugin/src/SiLUPlugin.cpp -------------------------------------------------------------------------------- /weights/yolov5n.engine: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/weights/yolov5n.engine -------------------------------------------------------------------------------- /weights/yolov5n.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/weights/yolov5n.onnx -------------------------------------------------------------------------------- /weights/yolov5n.plugin.engine: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/weights/yolov5n.plugin.engine -------------------------------------------------------------------------------- /weights/yolov5n.plugin.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-lishuang/yolov5_tensorrt/HEAD/weights/yolov5n.plugin.onnx --------------------------------------------------------------------------------