使用netron对TensorFlow、Pytorch、Keras、PaddlePaddle、MXNet、Caffe、ONNX、UFF、TNN、ncnn、OpenVINO等模型的可视化

欢迎大家关注笔者,你的关注是我持续更博的最大动力


原创文章,转载告知,盗版必究


使用netron对TensorFlow、Pytorch、Keras、PaddlePaddle、MXNet、Caffe、ONNX、UFF、TNN、ncnn、OpenVINO等模型的可视化

文章目录:


Netron是一个神经网络、深度学习和机器学习预训练模型的查看器或可视化工具


先介绍几个Netron的文档相关信息:


1 Netron支持的深度学习框架的预训练模型

Netron几乎支持可视化所有的主流的深度学习框架预训练模型,真的特别特别多,下面我们就开始掰一掰:

序号深度学习框架名深度学习框架的预训练模型后缀名
1TensorFlow.pb, .meta, .pbtxt, .ckpt, .index
2TensorFlow Lite.tflite
3TensorFlow.jsmodel.json, .pb
4Keras, .keras
5Pytorch.pt, .pth
6Torch.t7
7TorchScript.pt, .pth
8Caffe.caffemodel
9Caffe2predict_net.pb
10MXNet.model, -symbol.json
11PaddlePaddle.zip, __model__
12Chainer.npz, .h5
13CNTK.model, .cntk
14ONNX.onnx, .pb, .pbtxt
15UFF.uff
16ncnn.param
17TNN.tnnproto
18BigDL.bigdl, .model
19Deeplearning4j.zip
20Arm NN.armnn
21MediaPipe.pbtxt
22ML.NET.zip
23MNN.mnn
24scikit-learn.pkl
25OpenVINO.xml
26Core ML.mlmodel
27Tengine.tmfile
28Barracuda.nn
29Darknet.cfg

2Netron可视化模型的方式

Netron可视化模型的方式也很多,主要有:

  • 在线可视化模型
  • 离线APP可视化模型
  • 使用Netron API可视化模型

2.1 在线可视化模型

Netron在线可视化模型地址

1、打开Netron地址,可以看到如下界面

在这里插入图片描述
2、点击Open Model上传一个预训练的模型即可可视化

如下我上传一个resnet50.onnx模型可视化结果如下:
在这里插入图片描述

从可视化话的结果可以看出如下信息

  • 模型的网络整体结构图
  • 使用的深度学习框架及其版本信息
  • 模型输入维度
  • 模型输出维度
  • 同时你也可以可视化的图片保存下来

2.2 离线APP可视化模型

我是在windows系统环境下测试,其他环境同理安装软甲即可!

1、下载windows 的Netron的exe软件包

Netron4.4.2下载地址

在这里插入图片描述
2、下载完直接安装,界面和上面在线的Netron一样,直接导入模型可视化即可
在这里插入图片描述

2.3 使用Netron API可视化模型

使用API可视化,首先需要使用pip安装Netron

1、安装Netron

pip install netron

2、直接使用start(pred_model)函数可视化

netron.start(pred_model)

然后默认会打开8080端口,会直接跳转到浏览器中,可视化,如果没有跳转你就自己把url粘贴到浏览器中即可!
在这里插入图片描述

还有一些其他函数的使用,自从查看具体使用;
在这里插入图片描述


在这里插入图片描述




在这里插入图片描述
♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠

已标记关键词 清除标记
相关推荐
©️2020 CSDN 皮肤主题: 猿与汪的秘密 设计师:白松林 返回首页