yolov5模型直接导出torchscript格式、onnx格式

最近在学习训练yolov5模型,但在导出onnx格式文件总是出错,参考了很多文章都是采用直接运行export.py文件,但我确只导出了torchscript格式文件

 python export.py --weights runs/train/exp/weights/best.pt

我运行以上代码,导出的只有 torchscript格式文件,并没有其他人写的onnx格式文件,最后我上github的查看官方的文档,发现导出语句已经进行更改如下

python export.py --weights yolov5s-seg.pt --include onnx engine --img 640 --device 0

我自己进行尝试:

python export.py --weights runs/train/exp/weights/best.pt --include onnx engine

成功解决问题导出了best.onnx文件!再此记录一下!

  • 9
    点赞
  • 13
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值