Py学习  »  机器学习算法

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

深度学习与神经网络 • 5 年前 • 1503 次点击  

点击上方“深度学习与神经网络”,选择“置顶公众号”

重磅干货,第一时间送达


前言


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


若喜欢“深度学习与神经网络”推送的文章,请扫描下方二维码关注公众号!



您好,欢迎关注深度学习与神经网络。本公众号会定期推送深度学习的教程以及人工智能最新发展情况。感谢大家的支持!
回复“教程”可以看到最新的深度学习教程。
回复“加群”可以加入深度学习交流群。
投稿或合作请加微信:sdxxqbf


今天看啥 - 高品质阅读平台
本文地址:http://www.jintiankansha.me/t/7JRDbUI6Zt
Python社区是高质量的Python/Django开发社区
本文地址:http://www.python88.com/topic/22712
 
1503 次点击