opencv 曲线拟合

在OpenCV中，可以使用最小二乘法进行曲线拟合。下面是两种常见的曲线拟合方法：

1. 使用polyfit函数进行曲线拟合（引用）：

import numpy as np
import cv2

# 定义数据点
x = np.array([0, 1, 2, 3, 4, 5])
y = np.array([0, 0.8, 0.9, 0.1, -0.8, -1])

# 进行曲线拟合
z = np.polyfit(x, y, 3)  # 3表示拟合多项式的阶数

# 生成拟合曲线的x坐标
x_fit = np.linspace(0, 5, 100)

# 计算拟合曲线的y坐标
y_fit = np.polyval(z, x_fit)

# 绘制原始数据点和拟合曲线
plt.plot(x, y, 'o', label='Original Data')
plt.plot(x_fit, y_fit, label='Fitted Curve')
plt.legend()
plt.show()

2. 使用cv2.fitLine函数进行直线拟合（引用）：

import cv2
import numpy as np

# 定义数据点
points = np.array([[0, 0], [1, 1], [2, 2], [3, 3]], dtype=np.float32)

# 进行直线拟合
vx, vy, x, y = cv2.fitLine(points, cv2.DIST_L2, 0, 0.01, 0.01)

# 计算直线上的两个点
lefty = int((-x * vy / vx) + y)
righty = int(((cols - x) * vy / vx) + y)

# 绘制拟合直线
cv2.line(img, (cols - 1, righty), (0, lefty), (0, 255, 0), 2)