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

前言

前两天,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

原文发布于微信公众号 - CVer(CVerNews)

原文发表时间:2018-08-14

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏人工智能头条

谷歌 TensorFlow 物理检测 API,目前最好的物体识别方案?

1812
来自专栏企鹅号快讯

机器视觉:用图像驱动智能小车

本篇大致探索下图像的识别。实现了颜色识别,以及利用直方图来计算图像相似度。先来个视频。 摄像头实时捕捉画面,从画面中解析出蓝色的区域,并与预设图像对比,相似度小...

4980
来自专栏人工智能头条

如何确保机器学习最重要的起始步骤"特征工程"的步骤一致性?

关于特征工程,业界有这么一句话:数据和特征决定了机器学习的上限,而模型和算法只是逼近这个上限。

722
来自专栏大数据文摘

干脆面君,你给我站住!你已经被TensorFlow盯上了

1993
来自专栏目标检测和深度学习

用好这几款炼丹利器,调参从此得心应手 | Let's Fork

#实现显示结果的CV开发环境 ? Live CV 是一个用于实现显示结果的计算机视觉算法开发环境,基于 QtQuick 开发。它可以用于交互、链接和调整算法,以...

3758
来自专栏量子位

无需写代码!谷歌推出机器学习模型分析神器,代号What-If

今天,谷歌推出了已开源的TensorFlow可视化工具TensorBoard中一项新功能:What-If Tool,用户可在不编写程序代码的情况下分析机器学习(...

1243
来自专栏IT派

TensorFlow轻度入门

导语:虽然已经发过很多篇关于Tensorflow的入门推文了,不过我觉得有的读者应该是处于门外汉的状态,何不趁此就开始自己动手走一遍呢?另外再次感谢支持小编上课...

3834
来自专栏数据小魔方

think-cell chart系列9——折线图

今天跟大家分享think-cell chart系列的第9篇——折线图。 折线图是平时用的频率比较高的图表类型的了,下面教大家怎么在think-cell char...

5287
来自专栏人工智能

TensorFlow核心使用要点

正文之前,小梦先来说说什么是TensorFlow。TensorFlow是谷歌研发的第二代人工智能学习系统,可被用于语音识别或图像识别等多项机器深度学 习领域。T...

2367
来自专栏杨熹的专栏

详解 TensorBoard-如何调参

什么是 TensorBoard TensorBoard 是 TensorFlow 上一个非常酷的功能,我们都知道神经网络很多时候就像是个黑盒子,里面到底是什么样...

6728

扫码关注云+社区

领取腾讯云代金券