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

Posted CVer

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了开源 | 深度学习网络模型(model)可视化开源软件Netron相关的知识,希望对你有一定的参考价值。

重磅干货,第一时间送达


前言


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


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


Netron


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


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

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


来,让我们瞧一瞧!


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


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


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


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


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


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


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


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


下面是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

欢迎转发和点赞


以上是关于开源 | 深度学习网络模型(model)可视化开源软件Netron的主要内容,如果未能解决你的问题,请参考以下文章

微软开源 MMdnn:深度学习框架随心切换

机器学习 | 优秀Tensorflow开源项目汇总(上)

陈程:MMdnn如何让模型在不同深度学习框架间转换?

资源 | 小米开源移动端深度学习框架MACE:可转换TensorFlow模型

微软开源项目 NeuronBlocks:像搭积木一样构建 NLP 深度学习模型!

开源深度学习模型部署工具箱MMDeploy简介及安装