加载模型预训练权重时一般需要查看各节点信息,因此需要对权重文件进行可视化展示。
目前的Netron支持主流各种框架的模型结构可视化工作
框架 | 对应文件 |
---|---|
ONNX | .onnx, .pb |
Keras | .h5, .keras |
CoreML | .mlmodel |
TensorFlow Lite | .tflite |
安装
Netron支持windows,Linux,mac系统下载链接如下:
https://github.com/lutzroeder/netron/releases/tag/v4.7.0
我的电脑为windows系统,所以直接下载.exe文件即可
下载完成后直接双击进行安装。
使用
双击打开安装好的软件,然后选择点击file选择自己所要观看的模型文件即可
得到的网络结构图如下
参考博客:
https://blog.csdn.net/leviopku/article/details/81980249