前言

提示：Netron pth文件转onnx并显示出shape(size)的网络结构图。

一、pth转onnx

提示：pth文件是pytorch训练后保存的网络模型

开放式神经网络交换（Open Neural Network Exchange）ONNX格式
所谓开放就是ONNX定义了一组和环境，平台均无关的标准格式，可使模型在不同框架之间进行转移。

换句话说，无论你使用何种训练框架训练模型（比如TensorFlow/Pytorch/OneFlow/Paddle），在训练完毕后你都可以将这些框架的模型统一转换为ONNX这种统一的格式进行存储。注意ONNX文件不仅仅存储了神经网络模型的权重，同时也存储了模型的结构信息以及网络中每一层的输入输出和一些其它的辅助信息。

其实pth文件也可以直接用netron展示，但是pth文件展示模型网络结构有较多缺陷（因为模型通常只保存了网络参数，故展示时会变成下面这样没有网络结构）
eg

torch.onnx.export的各参数解析链接
onnx格式模型图片

有了网络结构，但是网络各层的shape没有

二、onnx格式给网络结构增加shape功能

代码如下（示例）：

增加节点shape信息的onnx格式模型在netron中的结构显示

Netron中显示

总结

提示：pytorch模型在netron中显示网络结构和节点shape的路径为：
.pth—>.onnx–>增加节点shape信息–>netron