神经网络可视化工具netron_netron可视化神经网络

可视化神经网络有两种方式：
1.网页版
https://lutzroeder.github.io/netron/

打开网页上传你的模型即可看到模型的结构。
2.python版
附上github链接：https://github.com/lutzroeder/netron
安装netron：pip install -i https://pypi.douban.com/simple/ netron

import netron
model_path = "../weights/ssd.onnx"
netron.start(model_path)
1
2
3

运行上述代码会自动跳转到http://localhost:8080
即可查看模型的结构。