0
点赞
收藏
分享

微信扫一扫

yolov5 tensorrt 推理实战

GG_lyf 2023-01-25 阅读 123


第一步 下载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
网上有很多转换方式,


举报

相关推荐

0 条评论