查看单个帖子
旧 2019-12-14, 20:13   #1
poster
高级会员
 
注册日期: 2019-11-21
帖子: 3,006
声望力: 66
poster 正向着好的方向发展
帖子 用Python或C语言编写的Matlab / Octave bwdist()

有人知道Matlab / Octave bwdist()函数的Python替代品吗?对于给定的矩阵,此函数将每个像元的欧几里德距离返回到最接近的非零像元。我看到了一个Octave C实现,一个纯Matlab实现,我想知道是否有人必须在ANSI C(不包含任何Matlab / Octave标头,因此我可以很容易地从Python集成)中实现它。

我提到的两个链接如下:

C ++

Matlab M文件

作为测试,Matlab代码/输出如下所示:

bw= [0 1 0 0 0; 1 0 0 0 0; 0 0 0 0 1; 0 0 0 0 0; 0 0 1 0 0] D = bwdist(bw) D = 1.00000 0.00000 1.00000 2.00000 2.00000 0.00000 1.00000 1.41421 1.41421 1.00000 1.00000 1.41421 2.00000 1.00000 0.00000 2.00000 1.41421 1.00000 1.41421 1.00000 2.00000 1.00000 0.00000 1.00000 2.00000 我在Python中测试了一个推荐的distance_transform_edt调用,它给出了以下结果:

从scipy导入ndimage导入numpy作为np

a = np.array(([0,1,0,0,0], [1,0,0,0,0], [0,0,0,0,1], [0,0,0,0,0], [0,0,1,0,0])) res = ndimage.distance_transform_edt(a) print res [[ 0. 1. 0. 0. 0.] [ 1. 0. 0. 0. 0.] [ 0. 0. 0. 0. 1.] [ 0. 0. 0. 0. 0.] [ 0. 0. 1. 0. 0.]] 此结果似乎与Octave / Matlab输出不匹配。



回答:

Matlab bwdist将距离返回到最接近的非零像元,而Python distance_transform_edt返回距离“最接近的背景元素”? SciPy文档尚不清楚它被认为是“背景”吗?它后面有一些类型转换机器;实际上, 0是背景,非零是前景。

因此,如果我们有矩阵a :

>>> a = np.array(([0,1,0,0,0], [1,0,0,0,0], [0,0,0,0,1], [0,0,0,0,0], [0,0,1,0,0])) 然后要计算相同的结果,我们需要将零替换为零,并将零替换为零,例如考虑矩阵1-a :

>>> a array([[0, 1, 0, 0, 0], [1, 0, 0, 0, 0], [0, 0, 0, 0, 1], [0, 0, 0, 0, 0], [0, 0, 1, 0, 0]]) >>> 1 - a array([[1, 0, 1, 1, 1], [0, 1, 1, 1, 1], [1, 1, 1, 1, 0], [1, 1, 1, 1, 1], [1, 1, 0, 1, 1]]) 在这种情况下, scipy.ndimage.morphology.distance_transform_edt提供了预期的结果:

>>> distance_transform_edt(1-a) array([[ 1. , 0. , 1. , 2. , 2. ], [ 0. , 1. , 1.41421356, 1.41421356, 1. ], [ 1. , 1.41421356, 2. , 1. , 0. ], [ 2. , 1.41421356, 1. , 1.41421356, 1. ], [ 2. , 1. , 0. , 1. , 2. ]])

更多&回答...
poster 当前离线   回复时引用此帖