pytorch对神经网络的特征图的可视-结合matplotlib工具
特征图的可视化在前向传播中返回需要可视化的特征图:对返回的特征图进行显示结果展示
在前向传播中返回需要可视化的特征图:
def forward(self, x):
output=[]
layers = self.dla(x)
p3 = self.p3(layers[3])
p3 = self.p3_l2(p3)
output.append(p3)
p4 = self.p4(layers[4])
p4 = self.p4_l2(p4)
out
原创
2020-07-02 17:15:26 ·
1027 阅读 ·
1 评论