2d 脊峰检测 // 使用 OpenCV 从密度图图像中提取线段

Question

我有线位置的密度图。当可视化密度图图像时，可以观察到多条线，每条线由一个或多个线段组成。我正在寻找一种解决方案，可以让我提取这些线并将它们表示为点列表（例如 [start_point, p2, p3, end_point]）。

是否有现有的 Python OpenCV (cv2) 函数或开源实现可以帮助我实现这一目标？我将不胜感激任何指导或建议。

显示线位置的密度图。 （即：我有什么）

线宽为1像素绘制的线段。 （即：我想要什么）

我想到的一种方法是蛮力法，我会遍历所有 x，然后遍历所有 y，以检测每个线段的“驼峰”，并将它们与前一个 x 附近的驼峰连接起来。然后，我将合并接近线性的段。我会重复这个过程，遍历所有 y，然后遍历 x，以检测垂直线。但是，如果可能的话，我正在寻找更有效的解决方案。

提前感谢您的帮助！