开源 | 深度学习网络模型(model)可视化开源软件Netron

2019-12-17 15:41:58 浏览数 (1)

前言

前两天,Amusi分享了一篇 经典卷积神经网络(CNN)结构可视化工具,该工具可用于可视化各种经典的卷积神经网络结构。如AlexNet、VGG-16、ResNet和YOLO等经典网络。

但上述工具的局限性很大,因为如果没有提供相关的prototxt文件,那么可视化结果就无从谈起。而且...Caffe现在热度也一般般了。

Netron

不好意思,Netron 这款工具真的可以为所欲为。Netron是微软家的lutzroeder小哥哥开源的一款软件,目前在github上已经有1300 人star(Amusi也有幸戳了一下)。这里Amusi有个疑问,国内咋没有人分享这个工具,难道......没错,是俺发现的早(羞羞.jpg)。

吹这么多,到底这款软件有啥功能和特点呢?

来,让我们瞧一瞧!

Netron是神经网络,深度学习和机器学习模型的可视化工具(viewer)。

Netron 支持目前大多数主流深度学习框架的模型,如下所示:

  • ONNX(.onnx,.pb)
  • Keras(.h5,.keras)
  • CoreML(.mlmodel)
  • TensorFlow Lite(.tflite)
  • Netron对Caffe(.caffemodel)
  • Caffe2(predict_net.pb)
  • MXNet(.model,-symbol.json)
  • TensorFlow.js(model.json,.pb)
  • TensorFlow(.pb,.meta)

注:支持的深度学习框架真的很多啊!但但但,唯独缺了PyTorch模型文件,这是啥个意思?

Netron Windows端软件使用教程

话不多说,下面以Windows版本的Netron软件举例,来展示一下该软件的"硬实力"。

首先,下载Netron-Setup-1.9.9.exe。这里膜拜一下Netron软件的更新速度,1.9.9正是前两天发布的版本。

link: https://github.com/lutzroeder/Netron/releases

下载并安装好Netron后,打开可见Netron的界面相当简约

这里举两个示例:Caffe的(.caffemodel)和TensorFlow(.pb,.meta)

下面是VGG-16.caffemodel的部分可视化截图

下面是xxx.meta的部分可视化截图(xxx是哪个模型呢?你可以猜一下)

对了,Netron可以将结构图保存下来(.PNG)......太厉害了吧

福利

Netron is a viewer for neural network, deep learning and machine learning models. Netron supports ONNX (.onnx, .pb), Keras (.h5, .keras), CoreML (.mlmodel) and TensorFlow Lite (.tflite). Netron has experimental support for Caffe (.caffemodel), Caffe2 (predict_net.pb), MXNet (.model, -symbol.json), TensorFlow.js (model.json, .pb) and TensorFlow (.pb, .meta).

为了防止大家下载文件受网络限制,Amusi已经将Window版的Netron下载好了,后台回复:Netron

即可获得Netron最新版的可执行文件。

github:

https://github.com/lutzroeder/Netron

0 人点赞