netron的github地址:https://github.com/lutzroeder/Netron
- 使用netron可视化工具可以清晰地看见神经网络的各层输入名和输出名以及具体的网络结构。相比tensorboard它更加轻量化,而且支持各种框架:
- 各平台安装方式:
- 测试netron。新建一个test.py文件,并与xxx.onnx在同一个目录下:
import netron
modelPath = "xxx.onnx"
netron.start(modelPath)
在终端执行:
python test.py
这时候点击终端输出的网址就可以在浏览器查看网络结构了:
http://localhost:8080