pytorch - 如何从 PyTorch 可视化多通道特征？

Question

我几乎是 PyTorch 的新手

我的 conv 输出大小之一是 [1, 25, 8, 32] (25=channel, 8=height, 32=width)

我可以使用挤压并使其达到 [25, 8, 32]。

但我对 25 频道感到困惑。

当我想可视化 25 个通道的总和并制作一个 GRAYorRGB 图像（1or3x8x32）时，我该如何处理代码？

我可以使用 matplot 或 tensorboardX 进行可视化..

Answer 1

很难可视化具有 3 个以上通道的图像，并且不清楚 25 维空间中的特征向量实际上是什么样子。

最直接的方法是将您拥有的 8x32 特征图可视化为大小为 8x32 的单独的 25 个灰度图像。每张图像都将显示特定神经元/卷积滤波器/通道（这些都是等效的）对特定空间位置的输入的“敏感度”程度。

特征可视化有更复杂的方法，你可以在这篇博文中找到更多的细节。