当前位置:   article > 正文

yolov8 tensorRT注意事项_yolov8tensort导出需要注意什么

yolov8tensort导出需要注意什么

1下载模型或使用训练好的模型的pt文件。

2转化为onnx模型

3简化模型onnx-simplifier=4.17(4.19有问题)

yolo mode=export model=C:/TensorRT-8.4.2.4/bin/best.pt format=onnx dynamic=True    simplify=True

4转化为trt(下载TensorRT-8.4.2.4在bin目录下执行)

trtexec.exe   --onnx=yolov8n.onnx  --saveEngine=yolov8n.trt  --buildOnly --minShapes=images:1x3x640x640 --optShapes=images:4x3x640x640 --maxShapes=images:8x3x640x640
本文内容由网友自发贡献,转载请注明出处:【wpsshop博客】
推荐阅读
相关标签
  

闽ICP备14008679号