发生背景
最近想要把yolov5训练得到的.pt
文件转成.onnx
以便后续使用,记录一下当前最佳实践。
解决思路
参考网上的参考的大多数文章,有一个明显的不足,就是转化之后的onnx并没有将对应的标签并入(即:smoke——>class 0)
但是发现yolov5根目录下自带export.py文件用于将.pt文件转化成其他格式。经过实践,可以保持标签对应关系。
参考链接 yolov5官方教程
最佳实践
<1.修改文件参数>
修改export.py
文件参数,如下图
898行:修改输出格式如下
873行:修改yaml文件以及.pt文件所在目录等,黄色框内default改成True(为了实现动态图像尺寸的输入输出)
<2.验证>
import netron
netron.start("smoke_best.onnx")