博客
关于我
神经网络常见可视化方法
阅读量:230 次
发布时间:2019-02-28

本文共 427 字,大约阅读时间需要 1 分钟。

1.打印出网络的参数量 , 直接打出网络结构(不够直观)

   .

2. 利用可视化工具,直观显示网络结构

Netron是一款可视化网络模型的软件,目前的Netron支持主流各种框架的模型结构可视化工 

支持的文件类型如下:

点开链接后,按自己的系统直接点击下载就可以了,比如我用的win10,就按如下下载: 

安装后,直接把网络文件(如onnx)直接拖入就能显示了:

 

比如,我常用的就是pytorch,那么首先把pytorch转为onnx文件:

import torch.onnximport torchvision input = torch.randn(10, 3, 224, 224).cuda()model = torchvision.models.alexnet(pretrained=True).cuda()torch.onnx.export(model, input, "alexnet.onnx", verbose=True)

 再拖进去就显示啦!

转载地址:http://dgpi.baihongyu.com/

你可能感兴趣的文章
Mysql建立中英文全文索引(mysql5.7以上)
查看>>
mysql建立索引的几大原则
查看>>
Mysql建表中的 “FEDERATED 引擎连接失败 - Server Name Doesn‘t Exist“ 解决方法
查看>>
MySQL开源工具推荐,有了它我卸了珍藏多年Nactive!
查看>>
MySQL异步操作在C++中的应用
查看>>
MySQL引擎讲解
查看>>
Mysql当前列的值等于上一行的值累加前一列的值
查看>>
MySQL当查询的时候有多个结果,但需要返回一条的情况用GROUP_CONCAT拼接
查看>>
MySQL必知必会(组合Where子句,Not和In操作符)
查看>>
MySQL必知必会总结笔记
查看>>
MySQL快速入门
查看>>
MySQL快速入门——库的操作
查看>>
mysql快速复制一张表的内容,并添加新内容到另一张表中
查看>>
mysql快速查询表的结构和注释,字段等信息
查看>>
mysql怎么删除临时表里的数据_MySQL中关于临时表的一些基本使用方法
查看>>
mysql性能优化
查看>>
mysql性能优化学习笔记-存储引擎
查看>>
MySQL性能优化必备25条
查看>>
Mysql性能优化(1):SQL的执行过程
查看>>
Mysql性能优化(2):数据库索引
查看>>