onnx模型可视化方法

onnx（Open Neural Network Exchange，开放神经网络交换）格式,是一个用于表示深度学习模型的标准,可使模型在不同框架之间进行转移，在使用pytorch框架好模型后需要部署到边缘端，.pth不大适合部署边缘端环境，最好需要转为onnx框架模型，在转换为.onnx后方便查看是否转换正确，可以通过可视化onnx模型来确认。

可视化教程链接：https://github.com/onnx/tutorials/blob/master/tutorials/VisualizingAModel.md

首先需要安装好onnx，然后安装以下步骤进行转换：

1. 先安装两个库：

pip install pydot
pip install Graphviz

2. 生成dot文件

python onnx/tools/net_drawer.py --input <path to squeezenet.onnx> --output squeezenet.dot --embed_docstring

3.生成可视化文件squeezenet.svg

dot -Tsvg squeezenet.dot -o squeezenet.svg

双击或者使用浏览器即可打开文件squeezenet.svg

下图为转换好的yolov3-tiny的onnx格式文件。