python - 分别为每个 channel 对 3d 数组与 2d 内核进行卷积

Question

我有一个大小为 [c, n, m] 的矩阵，其中 c 是 channel 数； n 和m 是宽度和高度。在特定示例中，我有一个具有 1000 个 channel 的矩阵。我想分别为每个 channel 与大小为 a x a 的内核进行卷积。在我的示例中，内核大小为 3 x 3。 scipy 或 numpy 中是否有任何函数可以执行此类操作而不用循环遍历 channel ？

我找到了 scipy.ndimage.convolve 函数，但我认为我不能在不使用循环的情况下将该函数应用到这个问题上。

Answer 1

我认为你只需要让内核成为三维的。像这样的东西应该可以工作:

kernel = kernel[:, :, None]

如果 scipy.ndimage.convolve 不适用于 3D 阵列，您可以尝试 scipy.signal.convolve。