mxnet模型可视化

使用netron

解析mxnet模型文件,可视化模型整体单元,但无法观察通道shape

pip install netron -i https://pypi.tuna.tsinghua.edu.cn/simple

netron直接运行,添加json文件即可

使用Graphviz,可视化

安装：

sudo apt-get install graphviz

可视化代码:

import mxnet as mx

sym,arg_params,aux_params = mx.model.load_checkpoint("resnet-18",0)

a = mx.viz.plot_network(sym, shape={"data": (1, 3, 224, 224)}, node_attrs={"shape": 'rect', "fixedsize": 'false'})

a.render('resnet-18')

用Anconada安装graphviz无效,直接用pip安装,有效可用