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 索引