网络可视化工具netron

安装教程

为了更加清晰的了解算法网络结构，可以使用网络可视化工具netron

安装教程：这里直接引用大白老师的教程，很详细：

netron安装教程

增加输出图尺寸

使用netron打开相关onnx文件时候，只有输入尺寸信息，没有输出，还是不清晰，可以通过相关代码进行转化后：

import onnx
from onnx import shape_inference

model = r'.onnx'
#上一步保存好的onnx格式的模型路径
onnx.save(onnx.shape_inference.infer_shapes(onnx.load(model)), model)
#增加节点的shape信息

1
2
3
4
5
6
7
8

这样再打开onnx图，就有了更详细的输入尺寸信息，就像下面这样，很清晰。