pytorch以Mnist为例进行中间层特征图可视化
接上一篇文章,在得到训练模型后,进行加载模型后,对模型中间层特征进行提取并输出预测精度:方法参考《pytorch 提取卷积神经网络的特征图可视化》class FeatureExtractor(nn.Module): def __init__(self,submodule,extracted_layer): super(FeatureExtractor,sel...
原创
2019-12-23 15:23:47 ·
3646 阅读 ·
0 评论