TensorRT学习笔记--基于TensorRT部署YoloV3, YoloV5和YoloV8

目录

1--完整项目

2--模型转换

3--编译项目

4--序列化模型

5--推理测试

---

1--完整项目

以下以 YoloV8 为例进行图片和视频的推理，完整项目地址如下：https://github.com/liujf69/TensorRT-Demo

git clone https://github.com/liujf69/TensorRT-Demo.gitcd TRT_YoloV8

2--模型转换

cd yolov8python gen_wts.py

3--编译项目

mkdir buildcd build cmake .. # 需要更改 CMakeLists.txtmake

4--序列化模型

./yolov8 -s ../yolov8/weights/yolov8s.wts ./yolov8s.engine s

5--推理测试

# 推理图片
./yolov8 -d ./yolov8s.engine -i ../images g# 推理视频
./yolov8 -d ./yolov8s.engine -v ../videos/test1.avi g