最小二乘拟合

本文对最小二乘拟合直线中出现的问题，当直线垂直时参数无法求出，使用ax+by+c=0或者p=xcos(a)+ysin(a)计算量会增加，针对这种情况，本算做了修正可以判断直线垂直情况。

 

[cpp] view plain copy

1. //最小二乘法直线拟合ay = kx + b  
2. static int _rtCalculateLine(const RtPoint* src, const int num, float* k, float* b, int* a)  
3. {  
4.     if (NULL == src || num < 2 ) return -1;  
5.   
6.     int i = 0;  
7.     long x = 0, y = 0, xx = 0, xy = 0, yy = 0, tem = 0;  
8.   
9.     while(i < num)  
10.     {  
11.         x += (src + i)->x;     
12.         y += (src + i)->y;  
13.         xx += (src + i)->x * (src + i)->x;  
14.         xy += (src + i)->x * (src + i)->y;  
15.         yy += (src + i)->y * (src + i)->y;  
16.         i++;  
17.     }  
18.   
19.     tem = xx*num - x*x;  
20.     if(0 == tem)  
21.     {  
22.         tem = yy*num - y*y;  
23.         if( 0 == tem) return -1;  
24.           
25.         *k = (num*xy - x*y)/tem;  
26.         *b = -(x - (*k)*y)/num;  
27.         *a = 0;  
28.         return 0;  
29.     }  
30.   
31.     *k = (num*xy - x*y)/tem;  
32.     *b = (y - (*k)*x)/num;  
33.     *a = 1;  
34.       
35.     return 0;  
36. }

转载于:https://www.cnblogs.com/cl1024cl/p/6205277.html