python--使用distance_transform_edt生成trimap
distance_transform_edt介绍
distance_transform_edt是scipy库中的函数,调用方式为:
from scipy import ndimage
ndimage.morphology.distance_transform_edt()
作用是距离转换,计算图像中非零点到最近背景点(即0)的距离。举例如下:
下面展示一些 内联代码片
。
import numpy as np
from scipy import ndimage
mat = np.array(([1, 0, 0, 0, 1],
[1, 1, 1, 0, 1],
[0, 1, 1, 1, 0],
[0, 0, 0, 1, 1],
[1, 1, 1, 1, 1]))
dis = ndimage.morphology.distance_transform_edt(mat)
print(dis)
结果:
[[1. 0. 0. 0. 1. ]
[1. 1. 1. 0. 1. ]
[0. 1. 1. 1. 0. ]
[0. 0. 0. 1. 1. ]
[1. 1. 1. 1.41421356 2.