Python:查找 3D 数组(图像堆栈)中邻域的最高值

Question

我使用模板匹配算法来检测目标图像中给定对象的实例。该算法生成的热图如下所示:

到目前为止，我已经使用 skimage.feature.peak_local_max 来查找热图中的局部峰值，效果很好。

现在我想针对模板的不同旋转运行模板匹配，这将为我提供几个热图。像这些:

我已将这些图像堆叠在一起以创建 3D 数组，现在我想找到本地邻域中的最高值。例如，邻域定义为 15 像素宽和高(但包括所有旋转)。

3D 数组的形状为(高度、宽度、旋转)。 z 轴对应于模板的旋转。

编辑:我自己找到了一个解决方案(来自 markuscosinus 的有用评论)，并将在此发布此解决方案，以防有人遇到同样的问题:

我查看了 scipy.ndimage.filters.maximum_filter 的实现(可以找到 here 。我稍微调整了代码以处理我的数据形状并使用 footprint = np.ones((size_neighborhood, size_neighborhood, nr_rotations)).

Answer 1

我自己找到了一个解决方案(来自 markuscosinus 的有用评论)，并将在此发布此解决方案，以防有人遇到同样的问题:

我查看了 scipy.ndimage.filters.maximum_filter 的实现(可以找到 here 。我稍微调整了代码以处理我的数据形状并使用 footprint = np.ones((size_neighborhood, size_neighborhood, nr_rotations)).