netron可视化

1：netron简介

在很多项目中，经常会遇到各种网络模型，在学习的过程中，我们需要快速去了解这些结构，若只是单纯的看模型文件，很难了解其结构。

netron可视化工具，可以清晰的看看见每一层的输入输出，网络总体的架构，而且还能支持各种网络框架，不可或缺的一个工具。

2：安装

（1）浏览器直接使用

     Netron浏览器中直接输入：     Netron   打开此链接就可以直接使用

界面如下：

然后Open Model 打开模型文件即可

（2） 不同操作系统的安装使用

netron官方的Github链接：https://github.com/lutzroeder/netron

（1）window系统
打开官方Github链接，点击Download处下载.exe应用程序，即可使用

（2）Linux系统
第一种方式：
打开官方Github链接，点击Download处下载文件，并运行安装命令

 

 

第二种方式：
在终端命令下输入pip install netron
下载好后，在终端下输入 netron，在浏览器上输入 loaclhost:8080 即可

（3）Mac系统
打开官方Github链接，点击Download处下载文件，并运行安装命令

参考链接 

我感觉直接使用浏览器版本的就可以，没必要安装使用，

3：onnx安装以及转化为onnx文件

   在对yolov5进行展现结构时，发现netron并不能展现全部的网络结构，所以netron对于pt文件不友好，如果直接将pt文件直接放入netron，模型结构不能全部展现；

  所以需要将pt文件转化成onnx文件。

①：首先安装onnx库

pip install onnx>=1.7.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
# for ONNX export

②：安装coremltools库

pip install coremltools==4.0 -i https://pypi.tuna.tsinghua.edu.cn/simple # for CoreML export

③：yolov5为例，将pt文件转化为onnx文件

python models/export.py --weights weights/yolov5s.pt --img 640 --batch 1

转化之后，界面如下

 找到这个文件将其倒入netron，就可以完整的展现网络模型结构了。

 

小白学习中，如若侵权，请联系我删除，如若有错，还希望能够不吝赐教。