yolov5 tensorrt 推理实战

  • 第一步 下载yolov5
git  clone https://github.com/ultralytics/yolov5.git
  • 第二步 下载 yolov5s.pt
cd yolov5
wget https://github.com/ultralytics/yolov5/releases/download/v6.0/yolov5s.pt
  • 第三步
 python export.py --weights yolov5s.pt --include onnx
  • 第四步 将onnx 模型量化成tensorrt格式 .engine
网上有很多转换方式,
posted @ 2022-08-19 22:44  luoganttcc  阅读(11)  评论(0编辑  收藏  举报