python求矩阵的局部极大值_python – 获取2D数组中局部最大值的坐标超过某个值...

import numpy as np

import scipy

import scipy.ndimage as ndimage

import scipy.ndimage.filters as filters

import matplotlib.pyplot as plt

fname = '/tmp/slice0000.png'

neighborhood_size = 5

threshold = 1500

data = scipy.misc.imread(fname)

data_max = filters.maximum_filter(data, neighborhood_size)

maxima = (data == data_max)

data_min = filters.minimum_filter(data, neighborhood_size)

diff = ((data_max - data_min) > threshold)

maxima[diff == 0] = 0

labeled, num_objects = ndimage.label(maxima)

slices = ndimage.find_objects(labeled)

x, y = [], []

for dy,dx in slices:

x_center = (dx.start + dx.stop - 1)/2

x.append(x_center)

y_center = (dy.start + dy.stop - 1)/2

y.append(y_center)

plt.imshow(data)

plt.savefig('/tmp/data.png', bbox_inches = 'tight')

plt.autoscale(False)

plt.plot(x,y, 'ro')

plt.savefig('/tmp/result.png', bbox_inches = 'tight')

给定data.png：

上面的程序产生了threshold.png,阈值为1500.降低阈值以获取更多的局部最大值：

参考文献：