1. 程式人生 > >opencv學習——最小二乘法擬合直線

opencv學習——最小二乘法擬合直線

 最小二乘法擬合直線

概念:最小二乘法多項式直線擬合,根據給定的點,求出它的函式y=f(x),當然求得準確的函式是不太可能的,但是我們能求出它的近似曲線y=φ(x)

原理
假設有點  , I = 1,2,3,……n,求近似曲線y=φ(x),並且使得y=φ(x)與y=f(x)的平方偏差和最小,偏差

  

 

其中我們要找到一組最好的a b ,“最好的”就是要使選出的a b能使得所有的誤差達到最小化。

在此要注意以下,y=ax+b 這裡面的未知量是什麼,很自然的說法是x y,實際上並不是,我們不用去解這個x和y ,因為x和y已經是給定的值了,當我們在找這條直線的時候,我們實際上並不關心x的值有多好,我們要的就是a 和b這兩個變數,它們可以描述x和y之間的關係,我們就是在試圖找出那條最適合的直線所對應的a和b。


可以看到最小二乘法對各個變數求偏導,使得偏導值為0,即可得到最小值,因為e是關於a b的函式,導數為0的點必定是最小值,進入正題

 分別對 a b求偏導可以得到:


關於solve函式 ,可以在這個連結檢視用法:solve函式使用

接下來上程式碼

#include<iostream>
#include<opencv2\opencv.hpp>
using namespace std;
using namespace cv;


int main()
{
	vector<Point>points;
	//(27 39) (8 5) (8 9) (16 22) (44 71) (35 44) (43 57) (19 24) (27 39) (37 52)

	points.push_back(Point(27, 39));
	points.push_back(Point(8, 5));
	points.push_back(Point(8, 9));
	points.push_back(Point(16, 22));
	points.push_back(Point(44, 71));
	points.push_back(Point(35, 44));
	points.push_back(Point(43, 57));
	points.push_back(Point(19, 24));
	points.push_back(Point(27, 39));
	points.push_back(Point(37, 52));
	Mat src = Mat::zeros(400, 400, CV_8UC3);

	for (int i = 0;i < points.size();i++)
	{
		//在原圖上畫出點
		circle(src, points[i], 3, Scalar(0, 0, 255), 1, 8);
	}
	//構建A矩陣 
	int N = 2;
	Mat A = Mat::zeros(N, N, CV_64FC1);

	for (int row = 0;row < A.rows;row++)
	{
		for (int col = 0; col < A.cols;col++)
		{
			for (int k = 0;k < points.size();k++)
			{
				A.at<double>(row, col) = A.at<double>(row, col) + pow(points[k].x, row + col);
			}
		}
	}
	//構建B矩陣
	Mat B = Mat::zeros(N, 1, CV_64FC1);
	for (int row = 0;row < B.rows;row++)
	{

		for (int k = 0;k < points.size();k++)
		{
			B.at<double>(row, 0) = B.at<double>(row, 0) + pow(points[k].x, row)*points[k].y;
		}
	}
	//A*X=B
	Mat X;
	//cout << A << endl << B << endl;
	solve(A, B, X,DECOMP_LU);
	cout << X << endl;
	vector<Point>lines;
	for (int x = 0;x < src.size().width;x++)
	{				// y = b + ax;
		double y = X.at<double>(0, 0) + X.at<double>(1, 0)*x;
		printf("(%d,%lf)\n", x, y);
		lines.push_back(Point(x, y));
	}
	polylines(src, lines, false, Scalar(255, 0, 0), 1, 8);
	imshow("src", src);
	
	//imshow("src", A);
	waitKey(0);
	return 0;
}

可以看到如下結果