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

Posted 2021-08-02 Tina姐

先看一下效果

我使用的是mac端

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

1 了解Netron

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

支持的终端

支持的终端包括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.

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

如何使用

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

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

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

更多功能

左上角有一个菜单工具

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

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

存在的问题

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

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

文章持续更新，可以关注微信公众号【医学图像人工智能实战营】获取最新动态，一个关注于医学图像处理领域前沿科技的公众号。坚持已实践为主，手把手带你做项目，打比赛，写论文。凡原创文章皆提供理论讲解，实验代码，实验数据。只有实践才能成长的更快，关注我们，一起学习进步~

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

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

觉得写的不错的话最后，求点赞，评论，收藏。或者一键三连

---

1. https://github.com/lutzroeder/netron ↩︎

2. https://github.com/lutzroeder/netron/issues/236 ↩︎