计算python中心像素周围区域的平均值

时间:2015-10-21 06:41:25

标签: python filter average mask

我有一个非常大的2D数组,想要过滤掉一些小功能。因此,我想计算中心像素周围区域的平均值。如果此值低于某个阈值,则在掩码中将中心像素的值设置为零。是否有现成的python功能?为简单起见,假设输入是一个带有整数条目的5乘9的数组,平均值应该通过2×2掩码进行,阈值为7。

1 个答案:

答案 0 :(得分:0)

尝试以下代码(阈值设置为7,平均设置为2×2区域):

Python 3.5.0 (default, Sep 27 2015, 12:06:50) 
[GCC 4.9.2] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import numpy as np
>>> from scipy import signal
>>> a = np.array([[2,3,2,3,5,7,3,6,1],[2,3,2,7,3,1,3,6,2],[345,1345,45,2,6,1,0,1,1],[2,3,2,456,5,7,3,1,1],[2,789,2,8,5,7,3,6,1]])
>>> b = np.array([[0.25,0.25],[0.25,0.25]])
>>> c = signal.convolve2d(a,b,boundary='symm',mode='same')
>>> threshold = 7.
>>> a[c<np.ones([5,9])*threshold]=0
>>> a
array([[   0,    0,    0,    0,    0,    0,    0,    0,    0],
   [   0,    0,    0,    0,    0,    0,    0,    0,    0],
   [ 345, 1345,   45,    2,    0,    0,    0,    0,    0],
   [   2,    3,    2,  456,    5,    0,    0,    0,    0],
   [   0,  789,    2,    8,    5,    0,    0,    0,    0]])
>>>