可视化神经网络模型的工具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了。
在这里插入图片描述


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM