python - 用于骨架化的倒角距离变换

Question

Python中是否有实现使用倒角距离变换计算图​​像骨架(骨架化)的函数或库？

以下链接是倒角距离变换的示例: http://www.inf.u-szeged.hu/~palagyi/skel/chamfer34.gif

谢谢

Answer 1

你的问题表述不清楚。倒角距离是两条曲线或两个二值图像之间的距离

假设您有两条曲线。

• 曲线A
• 曲线B

计算倒角变换的最简单方法是将曲线 A 转换为 Distance Transform在图像中。然后使用这些距离计算曲线 A 中每个点与曲线 B 中各点之间的最近距离。

换句话说，两条曲线或二值图像之间最近点距离的总和。

示例代码

import numpy as np
import cv2

# for Chamfer Distance between two curves
p_a - n x 2 numpy array
p_b - n x 2 numpy array
image_shape - (h, w) tuple
def chamfer(p_a, p_b, image_shape):
    mask = np.ones(image_shape[:2], dtype=np.uint8) * 255
    mask[p_a[:, 1].astype(int), p_a[:, 0].astype(int)] = 0
    dist = cv2.distanceTransform(mask, cv2.DIST_L2, 3,  dstType=cv2.CV_32F)
    return dist[p_b[:, 1].astype(int), p_b[:, 0].astype(int)].sum()

 chamfer_dist = 0.5 * (chamfer(p_a, p_b, image_shape) + chamfer(p_b, p_a, image_shape)

另一个选择是使用 Hausdorff Distance这在某些方面被认为更好