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

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)   # 实例化网络

你可能感兴趣的:(计算机视觉,-,Opencv,强化学习等的,趣味小实验,pytorch,机器学习,深度学习)