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.
719 B
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.
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.