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 ·
893 阅读 ·
1 评论