yolov5-cls部署之onnx导出

本文旨在介绍说明yolov5自带的分类如何导出动态的batch的onnx。其中输出两种形式：

形式（1）：导出带softmax映射到概率的

形式（2）：导出不带softmax的，这个也是官方默认的方式

一、动态导出

我们先看下如何得到我们想要的仅有动态batch的导出,在export.py中修改代码如下：

 

导出的onnx可视化如下：

注：官方直接导出，命令行使用simplify也可以得到上述结果

二、获取带softmax的onnx

原始官方导出是不带softmax概率映射的，因此想要获得带softmax的onnx，可以有两种做法：

（1）修改源码