当前位置:   article > 正文

yolov5 tensorrt 推理实战_使用tensorrt对yolov5推理

使用tensorrt对yolov5推理
  • 第一步 下载yolov5
git  clone https://github.com/ultralytics/yolov5.git
  • 1
  • 第二步 下载 yolov5s.pt
cd yolov5
wget https://github.com/ultralytics/yolov5/releases/download/v6.0/yolov5s.pt
  • 1
  • 2
  • 第三步
 python export.py --weights yolov5s.pt --include onnx
  • 1
  • 第四步 将onnx 模型量化成tensorrt格式 .engine
网上有很多转换方式,
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/爱喝兽奶帝天荒/article/detail/1013374
推荐阅读
相关标签
  

闽ICP备14008679号