Pytorch 深度卷积模型的特征可视化

• 1. 模型构建与可视化
• • 1.1 确定当前模型各层名称
  • 1.2 模型构建
  • 1.3 模型训练
• 2. 训练过程可视化与特征图
• • 2.1 获取完整节点信息
  • 2.2 可视化
• 参考文献资料

1. 模型构建与可视化

1.1 确定当前模型各层名称

可视化模型的特征层需要打印各层的名称：

• 安装torchsummary包

conda activate pytorch
pip install torchsummary

• 打印各层名称

model = model.to(device)   # 实例化网络