Loading...
墨滴

Tina姐

2021/07/05  阅读:29  主题:凝夜紫

模型可视化神器

先来看一下效果

我使用的是mac端

只要把你保存的模型拖曳进去,即可 可视化模型!

了解Netron

Netron[1]是一个神经网络、深度学习和机器学习模型的可视化工具,可以在多个终端使用。

支持的终端

支持的终端包括macOS, Linux, Windows, 浏览器, Python端

  • 各终端的下载方法

前往github下载对应版本[2]

支持的框架

ONNX, TensorFlow Lite, Keras, Caffe, Darknet, ncnn, MNN, PaddlePaddle, Core ML, MXNet, RKNN, MindSpore Lite, TNN, Barracuda, Tengine, TensorFlow.js, Caffe2 and UFF

PyTorch, TensorFlow, TorchScript, OpenVINO, Torch, Vitis AI, Arm NN, BigDL, Chainer, CNTK, Deeplearning4j, MediaPipe, ML.NET and scikit-learn.

基本是对所有的框架保存的模型都适用

如何使用

按照前面提供的地址,下载好对应的版本,如果只想试验一下,可以直接在网页端操作。网址[3]

选择 open model 或者直接拖曳一个模型进去即可显示。

如果你自己没有模型文件,可在链接1中下载不同框架的模型

更多功能

左上角有一个菜单工具

可以查看模型的属性,输入输出,选择显示或不显示每一层的名字,放大缩小,水平显示还是垂直显示等等

比如查看第一个卷积层的更多信息

存在的问题

它确实支持很多不同类型的模型,但我在加载‘.pth’文件时存在一个问题: 虽然显示了不同的节点,但是节点与节点间没有相关联👇

作者给出的解决方案是:使用torch.onnx进行包装 .pth不保存完整的图形,而只保存顶级节点,因为其余节点是用Python模块编码的。使用.onnx将保存完整的图形回答链接[4]

参考资料

[1]

Netron链接: https://github.com/lutzroeder/netron

[2]

github下载链接: https://github.com/lutzroeder/netron#install

[3]

网页端地址: https://netron.app/

[4]

回答链接: https://github.com/lutzroeder/netron/issues/236

Tina姐

2021/07/05  阅读:29  主题:凝夜紫

作者介绍

Tina姐