├── CMakeLists.txt ├── README.md ├── images ├── bus.jpg └── zidane.jpg ├── include ├── block.h ├── calibrator.h ├── config.h ├── model.h ├── process.h └── utils.h ├── plugin ├── yololayer.cu └── yololayer.h ├── src ├── block.cpp ├── calibrator.cpp ├── gen_wts.py ├── main.cpp ├── model.cpp ├── process.cu └── utils.cpp └── weights └── classes.txt /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaocao-tian/yolov8_tensorrt/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaocao-tian/yolov8_tensorrt/HEAD/README.md -------------------------------------------------------------------------------- /images/bus.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaocao-tian/yolov8_tensorrt/HEAD/images/bus.jpg -------------------------------------------------------------------------------- /images/zidane.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaocao-tian/yolov8_tensorrt/HEAD/images/zidane.jpg -------------------------------------------------------------------------------- /include/block.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaocao-tian/yolov8_tensorrt/HEAD/include/block.h -------------------------------------------------------------------------------- /include/calibrator.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaocao-tian/yolov8_tensorrt/HEAD/include/calibrator.h -------------------------------------------------------------------------------- /include/config.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaocao-tian/yolov8_tensorrt/HEAD/include/config.h -------------------------------------------------------------------------------- /include/model.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaocao-tian/yolov8_tensorrt/HEAD/include/model.h -------------------------------------------------------------------------------- /include/process.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaocao-tian/yolov8_tensorrt/HEAD/include/process.h -------------------------------------------------------------------------------- /include/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaocao-tian/yolov8_tensorrt/HEAD/include/utils.h -------------------------------------------------------------------------------- /plugin/yololayer.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaocao-tian/yolov8_tensorrt/HEAD/plugin/yololayer.cu -------------------------------------------------------------------------------- /plugin/yololayer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaocao-tian/yolov8_tensorrt/HEAD/plugin/yololayer.h -------------------------------------------------------------------------------- /src/block.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaocao-tian/yolov8_tensorrt/HEAD/src/block.cpp -------------------------------------------------------------------------------- /src/calibrator.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaocao-tian/yolov8_tensorrt/HEAD/src/calibrator.cpp -------------------------------------------------------------------------------- /src/gen_wts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaocao-tian/yolov8_tensorrt/HEAD/src/gen_wts.py -------------------------------------------------------------------------------- /src/main.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaocao-tian/yolov8_tensorrt/HEAD/src/main.cpp -------------------------------------------------------------------------------- /src/model.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaocao-tian/yolov8_tensorrt/HEAD/src/model.cpp -------------------------------------------------------------------------------- /src/process.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaocao-tian/yolov8_tensorrt/HEAD/src/process.cu -------------------------------------------------------------------------------- /src/utils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaocao-tian/yolov8_tensorrt/HEAD/src/utils.cpp -------------------------------------------------------------------------------- /weights/classes.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaocao-tian/yolov8_tensorrt/HEAD/weights/classes.txt --------------------------------------------------------------------------------