二维高斯曲面拟合法求取光斑中心及算法的C++实现

(1)二维高斯去曲面拟合推导

一个二维高斯方程可以写成如下形式:

其中,G为高斯分布的幅值,,为x,y方向上的标准差,对式(1)两边取对数,并展开平方项,整理后为:

假如参与拟合的数据点有N个,则将这个N个数据点写成矩阵的形式为:A = B C,

其中:

A为N*1的向量,其元素为:

B为N*5的矩阵:

C为一个由高斯参数组成的向量:

(2)求解二维高斯曲线拟合

N个数据点误差的列向量为:E=A-BC,用最小二乘法拟合,使其N个数据点的均方差最小,即:

在图像数据处理时,数据量比较大,为减小计算量,将矩阵B进行QR分解,即:B=QR,分解后Q为一个N*N的正交矩阵,R为一个N*5的上三角矩阵,对E=A-BC进行如下推导:

由于Q为正交矩阵,可以得到:

令:

其中:S为一个5维列向量;T为一个N-5维列向量;R1为一个5*5的上三角方阵,则

上式中,当S = R1C时取得最小值,因此只需解出:

即可求出:

中的

这些参数,这里先给出:

这里:

(3)C++代码实现,算法的实现过程中由于涉及大量的矩阵运算,所以采用了第三方的开源矩阵算法Eigen,这里真正用于高斯拟合的函数是

bool GetCentrePoint(float& x0,float& y0)

关于Eigen的用法请参考:

#include "stdafx.h"#include "GaussAlgorithm.h"VectorXf m_Vector_A;MatrixXf m_matrix_B;int m_iN =-1;bool InitData(int pSrc[100][100], int iWidth, int iHeight){if (NULL == pSrc || iWidth <=0 || iHeight <= 0)return false;m_iN = iWidth*iHeight;VectorXf tmp_A(m_iN);MatrixXf tmp_B(m_iN, 5);int i =0, j=0, iPos =0;while(i<iWidth){ j=0;while(j<iHeight){tmp_A(iPos) = pSrc[i][j] * log((float)pSrc[i][j]);tmp_B(iPos,0) = pSrc[i][j] ;tmp_B(iPos,1) = pSrc[i][j] * i;tmp_B(iPos,2) = pSrc[i][j] * j;tmp_B(iPos,3) = pSrc[i][j] * i * i;tmp_B(iPos,4) = pSrc[i][j] * j * j;++iPos;++j;}++i;}m_Vector_A = tmp_A;m_matrix_B = tmp_B;}bool GetCentrePoint(float& x0,float& y0){if (m_iN<=0)return false;//QR分解HouseholderQR<MatrixXf> qr;qr.compute(m_matrix_B);MatrixXf R = qr.matrixQR().triangularView<Upper>();MatrixXf Q = qr.householderQ();//块操作,,获取向量或矩阵的局部VectorXf S;S = (Q.transpose()* m_Vector_A).head(5);MatrixXf R1;R1 = R.block(0,0,5,5);VectorXf C;C = R1.inverse() * S;x0 = -0.5 * C[1] / C[3];y0 = -0.5 * C[2] / C[4];return true;}

其中:

版权声明:本文为博主原创文章,未经博主允许不得转载。

如果说,罗马是一座厚重和凝固的堡垒,

二维高斯曲面拟合法求取光斑中心及算法的C++实现

相关文章:

你感兴趣的文章:

标签云: