netron是一个神经网络模型可视化工具,下面对它做一下简单的介绍。
一、使用方式
- windows系统下载.exe应用程序,Linux和macOS系统下载对应应的用程序即可,点击这里去下载。打开应用程序,直接添加模型即可查看网络结构。
- 直接使用pip install netron安装netron包,然后使用以下程序去调用。
import netron
modelPath = "model.pb"
netron.start(modelPath)
此时它会自动跳转到浏览器显示模型结构。
二、目前支持的框架及其模型格式