2

如何提高此功能的速度？

def foo(mri_data, radius):

    mask = mri_data.copy()

    ny = len(mri_data[0,:])
    nx = len(mri_data[:])

    for y in xrange(0, ny):
        for x in xrange(0, nx):
            if (mri_data[x-radius:x+radius,y-radius:y+radius] != 1.0).all():
                mask[x,y] = 0.0                    
    return mask.copy()

它以 numpy 数组的形式接收图像切片。遍历每个像素并测试围绕该像素的边界框。如果框中没有值等于 1，则我们通过将其设置为 0 来丢弃该像素。

有人告诉我我可以使用numpy.convolve，但我不确定这有什么关系。

编辑：图像值在二进制范围内，因此最小值为 0.0，最大值为 1.0。介于 ex 之间的值：0.767。

4

2 回答 2

3

可以滥用卷积的情况之一。我不会使用它，但边界否则很乏味......

from scipy.ndimage import convolve

not_one = (mri_data != 1.0) # are you sure you want to compare with float like that?!

conv = convolve(not_one, np.ones((2*radius, 2*radius)))
all_not_one = (conv == (2*radius)**2)

mask[all_not_one] = 0

真的应该做同样的事情......

于 2013-02-07T19:14:47.327 回答
3

您正在做的事情称为 abinary_dilation但您的代码中有一个小错误。具体来说，当 x, y 小于半径时，您会得到负索引。这些负数是使用 numpy 索引规则解释的，这不是您在此处索引时更想要的，在图像的两个边缘上给您错误的结果。

这是一些使用二进制膨胀来完成同样事情的代码，并修复了上述错误。

import numpy as np
from scipy.ndimage import binary_dilation

def foo(mri_data, radius):
    structure = np.ones((2*radius, 2*radius))
    # I set the origin here to match your code
    mask = binary_dilation(mri_data == 1, structure, origin=-1)
    return np.where(mask, mri_data, 0)
于 2013-02-07T20:10:16.157 回答