OpenCV最小二乘法拟合空间平面

输入一个三维点的数组 std::vectorcv::Point3f Points3ds;
找到一个平面
Z=Ax+By+C
根据最小二乘法,使各个点到这个平面的距离最近:
S=∑(Axi + Byi + C - Zi) 2
求使得S最小的ABC的数值
首先取得最小值时,对各参数偏导数为零。
{ ∂ S ∂ A = 0 ∂ S ∂ B = 0 ∂ S ∂ C = 0 \left\{\begin{array}{l}\frac{\partial S}{\partial A}=0 \\ \\ \frac{\partial S}{\partial B}=0 \\ \\\frac{\partial S}{\partial C}=0\end{array}\right. AS=0BS=0CS=0

求偏导:

{ A ∑ x i 2 + B ∑ y i x i + C ∑ x i = ∑ z i x i A ∑ x i y i + B ∑ y i 2 + C ∑ y i = ∑ z i y i A ∑ x i + B ∑ y i + C n = ∑ z i \left\{\begin{array}{l}A \sum x_{i}^{2}+B \sum y_{i} x_{i}+C \sum x_{i}=\sum z_{i} x_{i} \\ \\ A \sum x_{i} y_{i}+B \sum y_{i}{ }^{2}+C \sum y_{i}=\sum z_{i} y_{i} \\ \\ A \sum x_{i}+B \sum y_{i}+C n=\sum z_{i}\end{array}\right. Axi2+Byixi+Cxi=zixiAxiyi+Byi2+Cyi=ziyiAxi+Byi+Cn=zi
整理得

{ ∑ 2 ( A x i + B y i + C − z i ) x i = 0 ∑ 2 ( A x i + B y i + C − z i ) y i = 0 ∑ 2 ( A x i + B y i + C − z i ) = 0 \left\{\begin{array}{l}\sum 2\left(A x_{i}+B y_{i}+C-z_{i}\right) x_{i}=0 \\ \\ \sum 2\left(A x_{i}+B y_{i}+C-z_{i}\right) y_{i}=0 \\\\ \sum 2\left(A x_{i}+B y_{i}+C-z_{i}\right)=0\end{array}\right. 2(Axi+Byi+Czi)xi=02(Axi+Byi+Czi)yi=02(Axi+Byi+Czi)=0
写为矩阵方程

∣ ∑ x i 2 ∑ x i y i ∑ x i ∑ x i y i ∑ y i 2 ∑ y i ∑ x i ∑ y i n ∣ ⋅ ∣ A B C ∣ = ∣ ∑ z i x i ∑ z i y i ∑ z i ∣ \left|\begin{array}{ccc}\sum x_{i}^{2} & \sum x_{i} y_{i} & \sum x_{i} \\\\ \sum x_{i} y_{i} & \sum y_{i}^{2} & \sum y_{i} \\\\ \sum x_{i} & \sum y_{i} & n\end{array}\right| \cdot\left|\begin{array}{l} A \\\\ B \\\\C \end{array}\right|=\left|\begin{array}{c}\sum z_{i} x_{i} \\ \\\sum z_{i} y_{i} \\\\ \sum z_{i}\end{array}\right| xi2xiyixixiyiyi2yixiyinABC=zixiziyizi
矩阵方程A ⋅ \cdot x=b
A是参数矩阵,不是上面多项式的参数,b是向量
通解x=A-1 ⋅ \cdot b
得到参数向量
∣ A B C ∣ \left|\begin{array}{l} A \\ B \\C \end{array}\right| ABC
代码如下:

void CaculateLaserPlane(std::vector<cv::Point3f> Points3ds,vector<double> &res)
{
	//最小二乘法拟合平面
	//获取cv::Mat的坐标系以纵向为x轴,横向为y轴,而cvPoint等则相反
	//系数矩阵
	cv::Mat A = cv::Mat::zeros(3, 3, CV_64FC1);
	//
	cv::Mat B = cv::Mat::zeros(3, 1, CV_64FC1);
	//结果矩阵
	cv::Mat X = cv::Mat::zeros(3, 1, CV_64FC1);
	double x2 = 0, xiyi = 0, xi = 0, yi = 0, zixi = 0, ziyi = 0, zi = 0, y2 = 0;
	for (int i = 0; i < Points3ds.size(); i++)
	{
		x2 += (double)Points3ds[i].x * (double)Points3ds[i].x;
		y2 += (double)Points3ds[i].y * (double)Points3ds[i].y;
		xiyi += (double)Points3ds[i].x * (double)Points3ds[i].y;
		xi += (double)Points3ds[i].x;
		yi += (double)Points3ds[i].y;
		zixi += (double)Points3ds[i].z * (double)Points3ds[i].x;
		ziyi += (double)Points3ds[i].z * (double)Points3ds[i].y;
		zi += (double)Points3ds[i].z;
	}
	A.at<double>(0, 0) = x2;
	A.at<double>(1, 0) = xiyi;
	A.at<double>(2, 0) = xi;
	A.at<double>(0, 1) = xiyi;
	A.at<double>(1, 1) = y2;
	A.at<double>(2, 1) = yi;
	A.at<double>(0, 2) = xi;
	A.at<double>(1, 2) = yi;
	A.at<double>(2, 2) = Points3ds.size();
	B.at<double>(0, 0) = zixi;
	B.at<double>(1, 0) = ziyi;
	B.at<double>(2, 0) = zi;
	//计算平面系数
	X = A.inv() * B;
	//A
	res.push_back(X.at<double>(0, 0));
	//B
	res.push_back(X.at<double>(1, 0));
	//Z的系数为1
	res.push_back(1.0);
	//C
	res.push_back(X.at<double>(2, 0));
	return;
}
  • 2
    点赞
  • 17
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
OpenCV中提供了函数fitLine()来实现最小二乘法拟合直线。 使用方法如下: 1. 导入必要的库和模块: ```python import cv2 import numpy as np ``` 2. 准备数据: 我们需要提供一些点的坐标,用于拟合直线。这些点可以是图像中的像素点,也可以是其他任意坐标系中的点。 ```python # 生成一些随机点 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度。 ```python # 使用fitLine()函数拟合直线 vx, vy, x, y = cv2.fitLine(points, cv2.DIST_L2, 0, 0.01, 0.01) ``` 4. 绘制直线: 得到直线的参数后,我们可以使用这些参数来绘制直线。可以使用cv2.line()函数来实现。 ```python # 绘制直线 lefty = int((-x*vy/vx) + y) righty = int(((200-x)*vy/vx)+y) cv2.line(img, (199,righty), (0,lefty), (0,255,0), 2) ``` 完整代码如下: ```python 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() ``` 运行结果如下图所示: ![拟合直线结果](https://img-blog.csdnimg.cn/20210622152054570.png)

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值