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.