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.git

cd TRT_YoloV8

2--模型转换

cd yolov8

python gen_wts.py

3--编译项目

mkdir build

cd build 

cmake .. # 需要更改 CMakeLists.txt

make

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

# 推理摄像头
./yolov8 -d ./yolov8s.engine -c 0 g # 0 表示 camera 索引