文章目录 前言 一、分步实现程序 1.引入库 2.定义网络加载模型 3.数据预处理 4. 载入数据 5.遍历模型中的layer及名字 6.依据选择的层,进行记录feature maps 二、完整程序如下 总结 前言 我们在做深度学习项目时,有时需要可视化网络模型中某层的特征图。本章将介绍一个可以可视化网络特征图的示例程序,仅供学习参考! 一、分步实现程序 1.引入库 代码如下(示例): # coding: utf-8 import os import torch import torchvision.utils as vutils import numpy as np from tensorboardX import SummaryWriter import torch.nn.functional as F imp