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.
31 lines
719 B
31 lines
719 B
# 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.
|
|
|