可视化神经网络模型的工具Netron

时间:2024-02-17 19:24:29

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了。
在这里插入图片描述