博客
关于我
神经网络常见可视化方法
阅读量: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 slave 停了_slave 停止。求解决方法
查看>>
MySQL SQL 优化指南:主键、ORDER BY、GROUP BY 和 UPDATE 优化详解
查看>>
MYSQL sql语句针对数据记录时间范围查询的效率对比
查看>>
mysql sum 没返回,如果没有找到任何值,我如何在MySQL中获得SUM函数以返回'0'?
查看>>
mysql Timestamp时间隔了8小时
查看>>
Mysql tinyint(1)与tinyint(4)的区别
查看>>
mysql union orderby 无效
查看>>
mysql v$session_Oracle 进程查看v$session
查看>>
mysql where中如何判断不为空
查看>>
MySQL Workbench 使用手册:从入门到精通
查看>>
mysql workbench6.3.5_MySQL Workbench
查看>>
MySQL Workbench安装教程以及菜单汉化
查看>>
MySQL Xtrabackup 安装、备份、恢复
查看>>
mysql [Err] 1436 - Thread stack overrun: 129464 bytes used of a 286720 byte stack, and 160000 bytes
查看>>
MySQL _ MySQL常用操作
查看>>
MySQL – 导出数据成csv
查看>>
MySQL —— 在CentOS9下安装MySQL
查看>>
MySQL —— 视图
查看>>
mysql 不区分大小写
查看>>
mysql 两列互转
查看>>