可视化神经网络模型的工具Netron
1、介绍
最近一直在研究神经网络,实现论文,搭建模型,有时候不清楚每层设置的参数,十分头疼。偶然发现了一个可视化模型的工具Netron,在windows,mac,linux上都可以直接安装,也支持web上浏览,十分方便,这里简单记录下。Windows安装版的如下图所示。
2、使用方法
web浏览器:https://lutzroeder.github.io/netron/
github地址:https://github.com/lutzroeder/netron
安装包:https://github.com/lutzroeder/netron/releases
除了以上使用方法,当然还可以配合python使用。
在anaconda下命令行安装pip install netron。
安装后,首先导入netron,再加载模型,netron.start(path)
import netron
import os
model_path = os.path.join(os.getcwd(),"model","train_iter_40000.caffemodel")
netron.start(model_path)
运行启动之后,在浏览器上输入http://localhost:8080/,就OK了。