【Pytorch学习笔记】可视化网络特征图featuremaps示例程序

---

前言

我们在做深度学习项目时，有时需要可视化网络模型中某层的特征图。本章将介绍一个可以可视化网络特征图的示例程序，仅供学习参考！

---

一、分步实现程序

1.引入库

代码如下（示例）：

# coding: utf-8
import os
import torch
import torchvision.utils as vutils
import numpy as np
from tensorboardX import SummaryWriter