博客
关于我
神经网络常见可视化方法
阅读量: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 分组统计SQL语句
查看>>
Mysql 分页
查看>>
Mysql 分页语句 Limit原理
查看>>
MySQL 创建新用户及授予权限的完整流程
查看>>
mysql 创建表,不能包含关键字values 以及 表id自增问题
查看>>
mysql 删除日志文件详解
查看>>
mysql 判断表字段是否存在,然后修改
查看>>
mysql 协议的退出命令包及解析
查看>>
mysql 取表中分组之后最新一条数据 分组最新数据 分组取最新数据 分组数据 获取每个分类的最新数据
查看>>
mysql 多个表关联查询查询时间长的问题
查看>>
mySQL 多个表求多个count
查看>>
mysql 多字段删除重复数据,保留最小id数据
查看>>
MySQL 多表联合查询:UNION 和 JOIN 分析
查看>>
MySQL 大数据量快速插入方法和语句优化
查看>>
mysql 如何给SQL添加索引
查看>>
mysql 字段区分大小写
查看>>
mysql 字段合并问题(group_concat)
查看>>
mysql 字段类型类型
查看>>
MySQL 字符串截取函数,字段截取,字符串截取
查看>>
MySQL 存储引擎
查看>>