深度学习模型可视化工具 Netron 使用教程
Netron
介绍
- Netron 是一个用于可视化机器学习模型、深度学习模型、神经网络、图模型（例如用于计算机视觉的 ONNX、Caffe、TensorFlow Lite、TensorFlow.js、Keras、Darknet、TVM、PyTorch、TorchScript、Core ML、ML.NET、NNEF、PaddlePaddle、OpenVINO、Arm NN等格式）的工具。它提供了一个用户友好的界面，允许用户浏览模型的结构和权重，并且支持各种深度学习框架。
- Netron 的特点包括：
- 跨平台：可以在浏览器、桌面（Windows、macOS、Linux）和命令行中使用。
- 交互式：用户可以通过图形界面交互地探索模型的结构和参数。
- 详细：可以查看模型的层、权重、激活函数等详细信息。
- 支持多种格式：支持多种深度学习模型格式，方便不同框架之间的转换和分析。
- Netron 的使用场景包括：
- 模型调试：帮助开发者理解模型的结构和行为。
- 模型转换：在不同的深度学习框架之间转换模型。
- 教育：作为教学工具，帮助学生理解深度学习模型的工作原理。
- 研究：分析和比较不同模型的结构和性能。
- 开源 GitHub 地址：https://github.com/lutzroeder/netron
发展历史
Netron 是由一个名为 “Lutz Roeder” 的开发者创建的，他是一位经验丰富的软件开发者和数据科学家。Lutz Roeder 也是著名的 .NET 反射浏览工具 “ILSpy” 的作者。
历史发展
- 初始版本：Netron 最初是作为一个命令行工具开发的，用于查看和分析深度学习模型文件。
- Web 版本：随着深度学习模型的普及，Lutz Roeder 决定开发一个 Web 版本，使得更多的人可以在浏览器中方便地查看模型。
- 扩展支持：随着时间的推移，Netron 不断扩展支持的模型格式，包括 ONNX、TensorFlow、PyTorch 等。
- 开源：Netron 的源代码在 GitHub 上开源，社区的贡献使得 Netron 不断改进和增加新功能。
主要里程碑
- 2017年：Netron 项目开始。
- 2018年：Netron Web 版本发布，支持 ONNX 格式。
- 2019年：支持更多格式，如 TensorFlow、Keras 等。
- 2020年：Netron 桌面应用发布，支持 Windows、macOS 和 Linux。
- 2021年：Netron 继续扩展支持的格式，包括 Core ML、ML.NET 等。
Netron使用方法
网页版
- 网址：https://netron.app/
- 界面比较简洁，拖入训练好的模型即可。
- 拖入一个模型 pt 文件， 可视化如下 ：
- 点击每个单元，展示具体卷积层类型和名字，继续点开，可以展现具体的数据，如下图：
- 将对应的 onnx 文件拖进去，可视化如下图：
本地安装使用
- 从GitHub 上找到对应安装包，下载安装：
2. 下载的文件如下，安装后的界面和网页版的差不多，如下：
导入库方法
- 也可以导入Netron库使用，如 GitHub 中介绍。
