如何模糊3D点阵,同时保持原始值? (Python)的

时间:2017-08-16 21:13:35

标签: python filter scipy gaussian gaussianblur

我有一个稀疏的3D数组值。我试图转动每个"点"通过对数组应用高斯滤波器,进入模糊的球体#34;

我希望点(x,y,z)处的原始值保持不变。我只是想围绕这一点创建衰减值......但是应用高斯滤波器也会改变原始(x,y,z)值。

我目前正在这样做:

dataCube = scipy.ndimage.filters.gaussian_filter(dataCube, 3, truncate=8)

有没有办法让我对此进行规范化,或者做一些事情以使我的原始值仍然在这个新的dataCube中?如果这不是最好的方法,我不一定要使用高斯滤波器。

1 个答案:

答案 0 :(得分:2)

您可以使用带有1作为其中心值的内核的卷积来执行此操作,并且宽度小于数据点之间的间距。

一维示例:

import numpy as np
import scipy.signal
data = np.array([0,0,0,0,0,5,0,0,0,0,0])
kernel = np.array([0.5,1,0.5])
scipy.signal.convolve(data, kernel, mode="same")

给出

array([ 0. ,  0. ,  0. ,  0. ,  2.5,  5. ,  2.5,  0. ,  0. ,  0. ,  0. ])

请注意,对于大型数组,fftconvolve可能要快得多。您还必须指定在数组边界应该发生的事情。

更新:三维示例

import numpy as np
from scipy import signal

# first build the smoothing kernel
sigma = 1.0     # width of kernel
x = np.arange(-3,4,1)   # coordinate arrays -- make sure they contain 0!
y = np.arange(-3,4,1)
z = np.arange(-3,4,1)
xx, yy, zz = np.meshgrid(x,y,z)
kernel = np.exp(-(xx**2 + yy**2 + zz**2)/(2*sigma**2))

# apply to sample data
data = np.zeros((11,11,11))
data[5,5,5] = 5.
filtered = signal.convolve(data, kernel, mode="same")

# check output
print filtered[:,5,5]

给出

[ 0.          0.          0.05554498  0.67667642  3.0326533   5.          3.0326533
  0.67667642  0.05554498  0.          0.        ]