前言:现在做这个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就行。