python - RANSAC算法查找图像的线参数并绘制线

Question

我有这两张图片:

我应该使用RANSAC算法找到它们的线参数并在它们上画出最好的线。

这是RANSAC算法的实现:

link

该算法的第一步是从原始数据中选择随机数据。如何使用图像作为输入？

Answer 1

这只是找到图像中所有非零点的问题。使用numpy.where查找所有非零坐标。

(y, x) = np.where(img)

img 是图像，假设它是灰度的。 y 是行位置，x 是列位置。您可以使用这些来输入帖子中看到的代码。