今天小编就为大家分享一篇pytorch 可视化feature map的示例代码，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

之前做的一些项目中涉及到feature map 可视化的问题，一个层中feature map的数量往往就是当前层out_channels的值，我们可以通过以下代码可视化自己网络中某层的feature map，个人感觉可视化feature map对调参还是很有用的。 不多说了，直接看代码： import torch from torch.autograd import Variable import torch.nn as nn import pickle from sys import pa