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

╰半夏微凉° 2024-05-24 05:47 82阅读 0赞

### 1、介绍 ###

最近一直在研究神经网络，实现论文，搭建模型，有时候不清楚每层设置的参数，十分头疼。偶然发现了一个可视化模型的工具Netron，在windows,mac,linux上都可以直接安装，也支持web上浏览，十分方便，这里简单记录下。Windows安装版的如下图所示。  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2JhaWR1XzMzMjU2MTc0_size_16_color_FFFFFF_t_70]

### 2、使用方法 ###

web浏览器：[https://lutzroeder.github.io/netron/][https_lutzroeder.github.io_netron]  
github地址：[https://github.com/lutzroeder/netron][https_github.com_lutzroeder_netron]  
安装包：[https://github.com/lutzroeder/netron/releases][https_github.com_lutzroeder_netron_releases]

除了以上使用方法，当然还可以配合python使用。  
在anaconda下命令行安装pip install netron。  
![在这里插入图片描述][20190329200155132.png]  
安装后，首先导入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了。  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2JhaWR1XzMzMjU2MTc0_size_16_color_FFFFFF_t_70 1]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2JhaWR1XzMzMjU2MTc0_size_16_color_FFFFFF_t_70]: https://image.dandelioncloud.cn/pgy_files/images/2024/05/24/54b012bf517241a69cc9eb94966125d2.png
[https_lutzroeder.github.io_netron]: https://lutzroeder.github.io/netron/
[https_github.com_lutzroeder_netron]: https://github.com/lutzroeder/netron
[https_github.com_lutzroeder_netron_releases]: https://github.com/lutzroeder/netron/releases
[20190329200155132.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/05/24/c63d2f7ce28f44a1a0d2feaf2643c923.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2JhaWR1XzMzMjU2MTc0_size_16_color_FFFFFF_t_70 1]: https://image.dandelioncloud.cn/pgy_files/images/2024/05/24/ff52e128547d4edeb2c9dd7e1fd1c6f7.png