pytorch 网络可视化(一)：torchsummary

1. 安装 torchsummary

在 Anaconda prompt 中进入自己的 pytorch 环境，安装依赖包。

pip install torchsummary

具体如下所示（其中 pytorch-cpu 是我自己的 pytorch 环境）:

2. 测试是否下载成功

等待安装完成后运行 python 进入交互式环境，导入 torchsummary, 不报错的话就是安装成功了。

3. 输出网络结构

完成以上步骤后，进入自己的 python 编辑环境，运行如下代码。

from torchsummary import summary
from torchvision.models import vgg16  # 以 vgg16 为例

myNet = vgg16()  # 实例化网络，可以换成自己的网络
summary(myNet, (3, 64, 64))  # 输出网络结构

运行代码，结果如下（VGG16）：

可以看出，torchsummary 不仅可以查看网络的顺序结构，还有网络参数量，网络模型大小等信息，非常实用。