在两端位于同一 x 坐标上的线之间画一条线

Question

我有带有线条的 png 文件作为 grascale png 上的掩码。 我只需要连接两端位于同一坐标上的线。 所以连接线是垂直的。 我也想在这里有一些公差，比方说 +/- 5 像素。

考虑下面的简单示例。 我只想将较细的线与一条垂直的附加线连接起来。 我不想连接较粗的线。

绘制线条的代码：

def lines():
    img = np.zeros((256,256,1), dtype=np.uint8)
    img = cv2.line(img, (50,50), (128 ,128), color=(255), thickness=1)
    img = cv2.line(img, (128 ,200), (240 ,250), color=(255), thickness=1)
    img = cv2.line(img, (96 ,54), (150,164), color=(255), thickness=2)

到目前为止，我有这样的事情。 仅用于连接点。

def dots():
    img = np.zeros((256,256,1), dtype=np.uint8)
    img = cv2.circle(img, (50,50), color=(255), radius=0, thickness=-1)
    img = cv2.circle(img, (128 ,200), color=(255), radius=0, thickness=-1)
    img = cv2.circle(img, (128 ,54), color=(255), radius=0, thickness=-1)

def connect_dots():
    img = dots()
    xy = np.argwhere(img == 255)
    x = xy == xy[:,1]
    x = np.argwhere(x[:,1] == True)
    x = xy[x]
    x = np.swapaxes(x, 1, 2)
    x = x[:,:2,:2]

    return x

def draw_line():
    img  = dots()
    start = np.flip(np.squeeze(connect_dots()[0,:,], axis=-1))
    end = np.flip(np.squeeze(connect_dots()[1,:,], axis=-1))
    print(start.shape, end)
    img = cv2.line(img, start, end, color=(177), thickness=1)
    cv2.imshow("connected lines", img)
    cv2.waitKey(0) 

但：

1. 我觉得完全是意大利面
2. 它只适用于点而不适用于线
3. 没有公差，该点必须完全位于相同的 x 坐标上，并且绘制的线必须完全垂直。

Answer 1

• 检测线的端点（它们在一侧有邻居但在另一侧没有）。

• 按横坐标对端点进行排序。

• 扫描端点列表以检测相距小于 5 像素的端点（除非它们是同一段的端点）。