You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
32 lines
719 B
32 lines
719 B
2 years ago
|
# Build TensorRT Engine By TensorRT Python API
|
||
|
|
|
||
|
|
When you want to build engine by API. You should generate the pickle weights parameters first.
|
||
|
|
|
||
|
|
``` shell
|
||
|
|
python3 gen_pkl.py -w yolov8s.pt -o yolov8s.pkl
|
||
|
|
```
|
||
|
|
|
||
|
|
You will get a `yolov8s.pkl` which contain the operators' parameters.
|
||
|
|
|
||
|
|
And you can rebuild `yolov8s` model in TensorRT api.
|
||
|
|
|
||
|
|
```
|
||
|
|
python3 build.py \
|
||
|
|
--weights yolov8s.pkl \
|
||
|
|
--iou-thres 0.65 \
|
||
|
|
--conf-thres 0.25 \
|
||
|
|
--topk 100 \
|
||
|
|
--fp16 \
|
||
|
|
--input-shape 1 3 640 640 \
|
||
|
|
--device cuda:0
|
||
|
|
```
|
||
|
|
|
||
|
|
***Notice !!!***
|
||
|
|
|
||
|
|
Now we only support static input shape model build by TensorRT api.
|
||
|
|
You'd best give the legal `input-shape`.
|
||
|
|
|
||
|
|
***Notice !!!***
|
||
|
|
|
||
|
|
Now we don't support YOLOv8-seg model building by API. It will be supported later.
|