yolov5的pt文件转换成各类型文件(onnx,tflite)

前言:现在做这个Android+Linux系统的板子硬件算法部署,pycharm中的权重文件是pt,但是这边要用qnn的文件,而我们要通过转换平台进行转换,目前只有onnx,tflite可以转成qnn,pt不能直接转换,于是在网上寻找转换方法。试了一天,最后竟然发现官方yolov5有转换文件,人麻了。。。虽然很简单的方法,但是还是记录一下吧,毕竟忙活一整天,哎,只能说自己还是太。。

首先打开yolov5官网进行下载

ultralytics/yolov5: YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite (github.com)

将文件下载下来,用pycharm打开,选择编译器,如果小伙伴们还没装,可以按照我之前安装pytorch环境进行安装,这样也更能保证后面运行成功

找到export.py文件,打开文件

找到这里,只需修改框住的三个部分即可,第一个框是你的数据集文件路径,第二个框是权重文件路径,第三个框是转换后的模型文件类型。(这个文件有错误是正常的,直接运行该文件就可以生成对应的目标文件)

如图所示生成对应类型的文件

要是想转成onnx就将tflite换成onnx就行。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值