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

作者：笔触狂放9 | 2024-02-22 10:54:52

踩

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博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/笔触狂放9/article/detail/129813