opencv最小二乘法拟合直线

OpenCV中提供了函数fitLine()来实现最小二乘法拟合直线。

使用方法如下：

1. 导入必要的库和模块：

import cv2
import numpy as np

2. 准备数据：

我们需要提供一些点的坐标，用于拟合直线。这些点可以是图像中的像素点，也可以是其他任意坐标系中的点。

# 生成一些随机点
points = np.random.randint(0, 200, (10, 2))

3. 拟合直线：

使用fitLine()函数来拟合直线，该函数需要传入一些参数：

- points：需要拟合的点的坐标。
- distType：点到直线的距离类型，可以选择CV_DIST_L1, CV_DIST_L2或CV_DIST_L12。
- param：累加器分辨率与rho的比值，一般取值为1。
- reps：在一条直线上的最小点数，一般取值为0。
- aeps：角度的最小精度，一般取值为0.01度。

# 使用fitLine()函数拟合直线
vx, vy, x, y = cv2.fitLine(points, cv2.DIST_L2, 0, 0.01, 0.01)

4. 绘制直线：

得到直线的参数后，我们可以使用这些参数来绘制直线。可以使用cv2.line()函数来实现。

# 绘制直线
lefty = int((-x*vy/vx) + y)
righty = int(((200-x)*vy/vx)+y)
cv2.line(img, (199,righty), (0,lefty), (0,255,0), 2)

完整代码如下：

import cv2
import numpy as np

# 生成一些随机点
points = np.random.randint(0, 200, (10, 2))

# 使用fitLine()函数拟合直线
vx, vy, x, y = cv2.fitLine(points, cv2.DIST_L2, 0, 0.01, 0.01)

# 绘制直线
img = np.zeros((200, 200, 3), dtype=np.uint8)
lefty = int((-x*vy/vx) + y)
righty = int(((200-x)*vy/vx)+y)
cv2.line(img, (199,righty), (0,lefty), (0,255,0), 2)

cv2.imshow("Fit Line", img)
cv2.waitKey(0)
cv2.destroyAllWindows()

运行结果如下图所示：