本文共 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/