可视化神经网络有两种方式:
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)
运行上述代码会自动跳转到http://localhost:8080
即可查看模型的结构。