0
点赞
收藏
分享

微信扫一扫

Netron 模型可视化神器,保存好的模型丢进去就能可视化!


先看一下效果

我使用的是mac端

Netron 模型可视化神器,保存好的模型丢进去就能可视化!_深度学习


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

1 了解Netron

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

支持的终端

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

  • 各终端的下载方法

前往github下载对应版本

支持的框架

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.

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

如何使用

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

Netron 模型可视化神器,保存好的模型丢进去就能可视化!_神经网络_02


选择 open model 或者直接拖曳一个模型进去即可显示。如果你自己没有模型文件,可在链接1中下载不同框架的模型

Netron 模型可视化神器,保存好的模型丢进去就能可视化!_pytorch_03

更多功能

左上角有一个菜单工具

Netron 模型可视化神器,保存好的模型丢进去就能可视化!_神经网络_04


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

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

Netron 模型可视化神器,保存好的模型丢进去就能可视化!_pytorch_05

存在的问题

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

Netron 模型可视化神器,保存好的模型丢进去就能可视化!_数据可视化_06


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

Netron 模型可视化神器,保存好的模型丢进去就能可视化!_pytorch_07

我是Tina, 我们下篇博客见~

白天工作晚上写文,呕心沥血

举报

相关推荐

0 条评论