博客
关于我
神经网络常见可视化方法
阅读量: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/

你可能感兴趣的文章
mysql5.7 for windows_MySQL 5.7 for Windows 解压缩版配置安装
查看>>
mysql5.7性能调优my.ini
查看>>
mysql5.7的安装和Navicat的安装
查看>>
MySQL8.0.29启动报错Different lower_case_table_names settings for server (‘0‘) and data dictionary (‘1‘)
查看>>
MySQL8修改密码报错ERROR 1819 (HY000): Your password does not satisfy the current policy requirements
查看>>
mysqlbinlog报错unknown variable ‘default-character-set=utf8mb4‘
查看>>
mysqldump 导出中文乱码
查看>>
mysqldump备份时忽略某些表
查看>>
mysqlreport分析工具详解
查看>>
MySQL一个表A中多个字段关联了表B的ID,如何关联查询?
查看>>
MYSQL一直显示正在启动
查看>>
MySQL一站到底!华为首发MySQL进阶宝典,基础+优化+源码+架构+实战五飞
查看>>
MySQL万字总结!超详细!
查看>>
Mysql下载以及安装(新手入门,超详细)
查看>>
mysql中cast() 和convert()的用法讲解
查看>>
mysql中floor函数的作用是什么?
查看>>
MySQL中group by 与 order by 一起使用排序问题
查看>>
mysql中having的用法
查看>>
mysql中int、bigint、smallint 和 tinyint的区别、char和varchar的区别详细介绍
查看>>
mysql中json_extract的使用方法
查看>>