如何在 python 中应用 2d 和 3d 过滤器

Question

我想在 python 中找到一个应用 2d 过滤器或 3d 过滤器的函数。该函数应该接收过滤函数和数据。 就像 Matlab 中的函数 filter2 和 imfilter， 或者像python中的函数scipy.ndimage.gaussian_filter()，它可以处理N维但仅适用于高斯过滤器（我希望过滤器成为一个参数）

我想我可以编写一个使用多维卷积 (scipy.ndimage.convolve()) 应用过滤器的函数，但是对于一维过滤器我没有成功。 我用于一维过滤的代码： signal.convolve(signal, filter_, mode='same', method='auto')，它给出的结果与函数 signal.filtfilt(filter_, 1, signal, padlen=len(signal)-1) 不同。

是否有应用 N-D 滤镜的功能？ 如果没有，我该如何使用卷积函数？

谢谢！

Answer 1

scipy.ndimage.generic_filter(input, function, size, footprint) 可能会有所帮助。

这里是文档scipy.ndimage.generic_filter 和用法tutorial。

您可以使用footprint 或size 来设计过滤器范围。例如size = 3或footprint = [[1, 1, 1], [1, 1, 1], [1, 1, 1]]设置以当前点为核心的3*3*3数组。 function 将其作为一维数组并返回您想要的标量。您可以将function(array) 定义为您的自定义过滤器。 泛型过滤器会用过滤器遍历input，默认用mode = 'reflect处理边界。