pytorch visualization of feature maps
发布时间
阅读量：
阅读量
在先前参与的多个项目中，曾遇到需要对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 path
path.append('/residual model path')
import residual_model
from residual_model import Residual_Model
model = Residual_Model()
model.load_state_dict(torch.load('./model.pkl'))
class myNet(nn.Module):
def __init__(self,pretraine全部评论 (0)
还没有任何评论哟~