博客
关于我
神经网络常见可视化方法
阅读量: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索引创建及使用注意事项
查看>>
mysql索引创建和使用注意事项
查看>>
MySQL索引原理以及查询优化
查看>>
Mysql索引合并(index merge)导致的死锁问题
查看>>
MySQL索引和查询优化
查看>>
mysql索引底层数据结构和算法
查看>>
Mysql索引底层结构的分析
查看>>
MySQL索引底层:B+树详解
查看>>
Mysql索引总结
查看>>
mysql索引最左匹配原则理解以及常见的sql使用的索引情况的实测
查看>>
Mysql索引类型
查看>>
MySQL索引背后的数据结构及算法原理
查看>>
mysql索引能重复吗_mysql “索引”能重复吗?“唯一索引”与“索引”区别是什么?...
查看>>
MySQL索引详解(IT枫斗者)
查看>>
MySQL索引那些事:什么是索引?为什么加索引就查得快了?
查看>>
Mysql索引(1):索引概述
查看>>
Mysql索引(2):索引结构
查看>>
Mysql索引(3):索引分类
查看>>
Mysql索引(4):索引语法
查看>>
mysql级联删除_Mysql笔记系列,DQL基础复习,Mysql的约束与范式
查看>>