卡尔曼滤波简介及其算法实现代码(C++/C/MATLAB)

来源:互联网 发布:淘宝护盾怎么授权 编辑:程序博客网 时间:2024/05/02 04:52
卡尔曼滤波器简介 

近来发现有些问题很多人都很感兴趣。所以在这里希望能尽自己能力跟大家讨论一些力所能及的算法。现在先讨论一下卡尔曼滤波器,如果时间和能力允许,我还希望能够写写其他的算法,例如遗传算法,傅立叶变换,数字滤波,神经网络,图像处理等等。

因为这里不能写复杂的数学公式,所以也只能形象的描述。希望如果哪位是这方面的专家,欢迎讨论更正。

卡尔曼滤波器 – Kalman Filter

一、什么是卡尔曼滤波器(What is the Kalman Filter?)

在学习卡尔曼滤波器之前,首先看看为什么叫“卡尔曼”。跟其他著名的理论(例如傅立叶变换,泰勒级数等等)一样,卡尔曼也是一个人的名字,而跟他们不同的是,他是个现代人!

卡尔曼全名Rudolf Emil Kalman,匈牙利数学家,1930年出生于匈牙利首都布达佩斯。1953,1954年于麻省理工学院分别获得电机工程学士及硕士学位。1957年于哥伦比亚大学获得博士学位。我们现在要学习的卡尔曼滤波器,正是源于他的博士论文和1960年发表的论文《A New Approach to Linear Filtering and Prediction Problems》(线性滤波与预测问题的新方法)。如果对这编论文有兴趣,可以到这里的地址下载: http://www.cs.unc.edu/~welch/media/pdf/Kalman1960.pdf。 

简单来说,卡尔曼滤波器是一个“optimal recursive data processing algorithm(最优化自回归数据处理算法)”。对于解决很大部分的问题,他是最优,效率最高甚至是最有用的。他的广泛应用已经超过30年,包括机器人导航,控制,传感器数据融合甚至在军事方面的雷达系统以及导弹追踪等等。近年来更被应用于计算机图像处理,例如头脸识别,图像分割,图像边缘检测等等。

二、卡尔曼滤波器的介绍(Introduction to the Kalman Filter)

为了可以更加容易的理解卡尔曼滤波器,这里会应用形象的描述方法来讲解,而不是像大多数参考书那样罗列一大堆的数学公式和数学符号。但是,他的5条公式是其核心内容。结合现代的计算机,其实卡尔曼的程序相当的简单,只要你理解了他的那5条公式。

在介绍他的5条公式之前,先让我们来根据下面的例子一步一步的探索。

假设我们要研究的对象是一个房间的温度。根据你的经验判断,这个房间的温度是恒定的,也就是下一分钟的温度等于现在这一分钟的温度(假设我们用一分钟来做时间单位)。假设你对你的经验不是100%的相信,可能会有上下偏差几度。我们把这些偏差看成是高斯白噪声(White Gaussian Noise),也就是这些偏差跟前后时间是没有关系的而且符合高斯分配(Gaussian Distribution)。另外,我们在房间里放一个温度计,但是这个温度计也不准确的,测量值会比实际值偏差。我们也把这些偏差看成是高斯白噪声。

好了,现在对于某一分钟我们有两个有关于该房间的温度值:你根据经验的预测值(系统的预测值)和温度计的值(测量值)。下面我们要用这两个值结合他们各自的噪声来估算出房间的实际温度值。

假如我们要估算k时刻的是实际温度值。首先你要根据k-1时刻的温度值,来预测k时刻的温度。因为你相信温度是恒定的,所以你会得到k时刻的温度预测值是跟 k-1时刻一样的,假设是23度,同时该值的高斯噪声的偏差是5度(5是这样得到的:如果k-1时刻估算出的最优温度值的偏差是3,你对自己预测的不确定度是4度,他们平方相加再开方,就是5)。然后,你从温度计那里得到了k时刻的温度值,假设是25度,同时该值的偏差是4度。

由于我们用于估算k时刻的实际温度有两个温度值,分别是23度和25度。究竟实际温度是多少呢?相信自己还是相信温度计呢?究竟相信谁多一点,我们可以用他们的 covariance来判断。因为Kg^2=5^2/(5^2+4^2),所以Kg=0.78,我们可以估算出k时刻的实际温度值是:23+0.78* (25-23)=24.56度。可以看出,因为温度计的covariance比较小(比较相信温度计),所以估算出的最优温度值偏向温度计的值。

现在我们已经得到k时刻的最优温度值了,下一步就是要进入k+1时刻,进行新的最优估算。到现在为止,好像还没看到什么自回归的东西出现。对了,在进入 k+1时刻之前,我们还要算出k时刻那个最优值(24.56度)的偏差。算法如下:((1-Kg)*5^2)^0.5=2.35。这里的5就是上面的k时刻你预测的那个23度温度值的偏差,得出的2.35就是进入k+1时刻以后k时刻估算出的最优温度值的偏差(对应于上面的3)。

就是这样,卡尔曼滤波器就不断的把covariance递归,从而估算出最优的温度值。他运行的很快,而且它只保留了上一时刻的covariance。上面的Kg,就是卡尔曼增益(Kalman Gain)。他可以随不同的时刻而改变他自己的值,是不是很神奇!

下面就要言归正传,讨论真正工程系统上的卡尔曼。

三、卡尔曼滤波器算法(The Kalman Filter Algorithm)

在这一部分,我们就来描述源于Dr Kalman 的卡尔曼滤波器。下面的描述,会涉及一些基本的概念知识,包括概率(Probability),随即变量(Random Variable),高斯或正态分配(Gaussian Distribution)还有State-space Model等等。但对于卡尔曼滤波器的详细证明,这里不能一一描述。

首先,我们先要引入一个离散控制过程的系统。该系统可用一个线性随机微分方程(Linear Stochastic Difference equation)来描述:

X(k)=A X(k-1)+B U(k)+W(k) 

再加上系统的测量值:

Z(k)=H X(k)+V(k) 

上两式子中,X(k)是k时刻的系统状态,U(k)是k时刻对系统的控制量。A和B是系统参数,对于多模型系统,他们为矩阵。Z(k)是k时刻的测量值,H 是测量系统的参数,对于多测量系统,H为矩阵。W(k)和V(k)分别表示过程和测量的噪声。他们被假设成高斯白噪声(White Gaussian Noise),他们的covariance 分别是Q,R(这里我们假设他们不随系统状态变化而变化)。

对于满足上面的条件(线性随机微分系统,过程和测量都是高斯白噪声),卡尔曼滤波器是最优的信息处理器。下面我们来用他们结合他们的covariances 来估算系统的最优化输出(类似上一节那个温度的例子)。

首先我们要利用系统的过程模型,来预测下一状态的系统。假设现在的系统状态是k,根据系统的模型,可以基于系统的上一状态而预测出现在状态:

X(k|k-1)=A X(k-1|k-1)+B U(k) ……….. (1)

式(1)中,X(k|k-1)是利用上一状态预测的结果,X(k-1|k-1)是上一状态最优的结果,U(k)为现在状态的控制量,如果没有控制量,它可以为0。

到现在为止,我们的系统结果已经更新了,可是,对应于X(k|k-1)的covariance还没更新。我们用P表示covariance:

P(k|k-1)=A P(k-1|k-1) A’+Q ……… (2)

式 (2)中,P(k|k-1)是X(k|k-1)对应的covariance,P(k-1|k-1)是X(k-1|k-1)对应的 covariance,A’表示A的转置矩阵,Q是系统过程的covariance。式子1,2就是卡尔曼滤波器5个公式当中的前两个,也就是对系统的预测。

现在我们有了现在状态的预测结果,然后我们再收集现在状态的测量值。结合预测值和测量值,我们可以得到现在状态(k)的最优化估算值X(k|k):

X(k|k)= X(k|k-1)+Kg(k) (Z(k)-H X(k|k-1)) ……… (3)

其中Kg为卡尔曼增益(Kalman Gain):

Kg(k)= P(k|k-1) H’ / (H P(k|k-1) H’ + R) ……… (4)

到现在为止,我们已经得到了k状态下最优的估算值X(k|k)。但是为了要另卡尔曼滤波器不断的运行下去直到系统过程结束,我们还要更新k状态下X(k|k)的covariance:

P(k|k)=(I-Kg(k) H)P(k|k-1) ……… (5)

其中I 为1的矩阵,对于单模型单测量,I=1。当系统进入k+1状态时,P(k|k)就是式子(2)的P(k-1|k-1)。这样,算法就可以自回归的运算下去。

卡尔曼滤波器的原理基本描述了,式子1,2,3,4和5就是他的5 个基本公式。根据这5个公式,可以很容易的实现计算机的程序。

下面,我会用程序举一个实际运行的例子

四、 简单例子(A Simple Example)

这里我们结合第二第三节,举一个非常简单的例子来说明卡尔曼滤波器的工作过程。所举的例子是进一步描述第二节的例子,而且还会配以程序模拟结果。

根据第二节的描述,把房间看成一个系统,然后对这个系统建模。当然,我们见的模型不需要非常地精确。我们所知道的这个房间的温度是跟前一时刻的温度相同的,所以A=1。没有控制量,所以U(k)=0。因此得出:

X(k|k-1)=X(k-1|k-1) ……….. (6)

式子(2)可以改成:

P(k|k-1)=P(k-1|k-1) +Q ……… (7)

因为测量的值是温度计的,跟温度直接对应,所以H=1。式子3,4,5可以改成以下:

X(k|k)= X(k|k-1)+Kg(k) (Z(k)-X(k|k-1)) ……… (8)

Kg(k)= P(k|k-1) / (P(k|k-1) + R) ……… (9)

P(k|k)=(1-Kg(k))P(k|k-1) ……… (10)

现在我们模拟一组测量值作为输入。假设房间的真实温度为25度,我模拟了200个测量值,这些测量值的平均值为25度,但是加入了标准偏差为几度的高斯白噪声(在图中为蓝线)。

为了令卡尔曼滤波器开始工作,我们需要告诉卡尔曼两个零时刻的初始值,是X(0|0)和P(0|0)。他们的值不用太在意,随便给一个就可以了,因为随着卡尔曼的工作,X会逐渐的收敛。但是对于P,一般不要取0,因为这样可能会令卡尔曼完全相信你给定的X(0|0)是系统最优的,从而使算法不能收敛。我选了 X(0|0)=1度,P(0|0)=10。

该系统的真实温度为25度,图中用黑线表示。图中红线是卡尔曼滤波器输出的最优化结果(该结果在算法中设置了Q=1e-6,R=1e-1)。

Kalman.jpg 

最佳线性滤波理论起源于 40 年代美国科学家 Wiener 和前苏联科学家K олмогоров 等人的研究工作,后人统称为维纳滤波理论。从理论上说,维纳滤波的最大缺点是必须用到无限过去的数据,不适用于实时处理。为了克服这一缺点, 60 年代 Kalman 把状态空间模型引入滤波理论,并导出了一套递推估计算法,后人称之为卡尔曼滤波理论。卡尔曼滤波是以最小均方误差为估计的最佳准则,来寻求一套递推估计的算法,其基本思想是:采用信号与噪声的状态空间模型,利用前一时刻地估计值和现时刻的观测值来更新对状态变量的估计,求出现时刻的估计值。它适合于实时处理和计算机运算。 

现设线性时变系统的离散状态防城和观测方程为: 

X(k) = F(k,k-1)*X(k-1)+T(k,k-1)*X(k-1)  
Y(k) = H(k)*X(k)+N(k) 

其中,X(k)和Y(k)分别是k时刻的状态矢量和观测矢量 
F(k,k-1)为状态转移矩阵 
U(k)为k时刻动态噪声 
T(k,k-1)为系统控制矩阵 
H(k)为k时刻观测矩阵 
N(k)为k时刻观测噪声 

则卡尔曼滤波的算法流程为: 

预估计X(k)^= F(k,k-1)*X(k-1)   

1.计算预估计协方差矩阵
C(k)^=F(k,k-1)×C(k)×F(k,k-1)'+T(k,k-1)×Q(k)×T(k,k-1)'
Q(k) = U(k)×U(k)'  

2.计算卡尔曼增益矩阵
K(k) = C(k)^×H(k)'×[H(k)×C(k)^×H(k)'+R(k)]^(-1)
R(k) = N(k)×N(k)'  

3.更新估计
X(k)~=X(k)^+K(k)×[Y(k)-H(k)×X(k)^]  

4.计算更新后估计协防差矩阵
C(k)~ = [I-K(k)×H(k)]×C(k)^×[I-K(k)×H(k)]'+K(k)×R(k)×K(k)'  

5.X(k+1) = X(k)~
C(k+1) = C(k)~

重复以上步骤 

五、MATLAB实现代码
  1. %%%% Constant Velocity Model Kalman Filter Simulation %%%%

  2. %==========================================================================

  3. clear all; close all; clc;
  4. %% Initial condition
  5. ts = 1; % Sampling time
  6. t = [0:ts:100];
  7. T = length(t);
  8. %% Initial state
  9. x = [0 40 0 20]'; 
  10. x_hat = [0 0 0 0]';
  11. %% Process noise covariance
  12. q = 5
  13. Q = q*eye(2);

  14. %% Measurement noise covariance
  15. r = 5
  16. R = r*eye(2);

  17. %% Process and measurement noise
  18. w = sqrt(Q)*randn(2,T);   % Process noise
  19. v = sqrt(R)*randn(2,T);   % Measurement noise
  20. %% Estimate error covariance initialization
  21. p = 5;
  22. P(:,:,1) = p*eye(4);

  23. %========================================================================== 

  24. %% Continuous-time state space model
  25. %{
  26. x_dot(t) = Ax(t)+Bu(t)
  27. z(t) = Cx(t)+Dn(t)
  28. %}
  29. A = [0 1 0 0;
  30.        0 0 0 0;
  31.        0 0 0 1; 
  32.        0 0 0 0];
  33. B = [0 0;
  34.        1 0;
  35.        0 0;
  36.        0 1];
  37. C = [1 0 0 0;
  38.        0 0 1 0];
  39. D = [1 0;
  40.        0 1];

  41. %% Discrete-time state space model
  42. %{
  43. x(k+1) = Fx(k)+Gw(k)
  44. z(k) = Hx(k)+Iv(k)
  45. Continuous to discrete form by zoh
  46. %}
  47. sysc = ss(A,B,C,D);
  48. sysd = c2d(sysc, ts, 'zoh');
  49. [F G H I] = ssdata(sysd);
  50. %% Practice state of target
  51. for i = 1:T-1
  52.     x(:,i+1) = F*x(:,i);
  53. end
  54. x = x+G*w;    % State variable with noise
  55. z = H*x+I*v; % Measurement value with noise

  56. %========================================================================== 

  57. %%% Kalman Filter 
  58. for i = 1:T-1
  59.     
  60. %% Prediction phase
  61.     x_hat(:,i+1) = F*x_hat(:,i);          
  62.     % State estimate predict
  63.     P(:,:,i+1) = F*P(:,:,i)*F'+G*Q*G';    
  64.     % Tracking error covariance predict
  65.     P_predicted(:,:,i+1) = P(:,:,i+1);    
  66.    
  67. %% Kalman gain
  68.     K = P(:,:,i+1)*H'*inv(H*P(:,:,i+1)*H'+R);
  69.     
  70. %% Updata step
  71.     x_hat(:,i+1) = x_hat(:,i+1)+K*(z(:,i+1)-H*x_hat(:,i+1));            
  72.     % State estimate update
  73.     P(:,:,i+1) = P(:,:,i+1)-K*H*P(:,:,i+1);                       
  74.     % Tracking error covariance update
  75.     P_updated(:,:,i+1) = P(:,:,i+1);                             
  76.    
  77. end

  78. %==========================================================================

  79. %% Estimate error
  80. x_error = x-x_hat;
  81. %% Graph 1 practical and tracking position
  82. figure(1)
  83. plot(x(1,:),x(3,:),'r');
  84. hold on;
  85. plot(x_hat(1,:),x_hat(3,:),'g.');
  86. title('2D Target Position')
  87. legend('Practical Position','Tracking Position')
  88. xlabel('X axis [m]')
  89. ylabel('Y axis [m]')
  90. hold off;
  91. %% Graph 2 
  92. figure(2) 
  93. plot(t,x(1,:)),grid on;
  94. hold on;
  95. plot(t,x_hat(1,:),'r'),grid on;
  96. title('Practical and Tracking Position on X axis')
  97. legend('Practical Position','Tracking Position')
  98. xlabel('Time [sec]')
  99. ylabel('Position [m]')
  100. hold off;
  101. %% Graph 3
  102. figure(3) 
  103. plot(t,x_error(1,:)),grid on;
  104. title('Position Error on X axis')
  105. xlabel('Time [sec]')
  106. ylabel('Position RMSE [m]')
  107. hold off;
  108. %% Graph 4
  109. figure(4)
  110. plot(t,x(2,:)),grid on;
  111. hold on;
  112. plot(t,x_hat(2,:),'r'),grid on;
  113. title('Practical and Tracking Velocity on X axis')
  114. legend('Practical Velocity','Tracking Velocity')
  115. xlabel('Time [sec]')
  116. ylabel('Velocity [m/sec]')
  117. hold off;
  118. %% Graph 5
  119. figure(5)
  120. plot(t,x_error(2,:)),grid on;
  121. title('Velocity Error on X axis')
  122. xlabel('Time [sec]')
  123. ylabel('Velocity RMSE [m/sec]')
  124. hold off;
  125. %==========================================================================
复制代码
六、C语言实现代码
  1. #include "stdlib.h"
  2. #include "rinv.c"
  3. int lman(n,m,k,f,q,r,h,y,x,p,g)
  4. int n,m,k;
  5. double f[],q[],r[],h[],y[],x[],p[],g[];
  6. { int i,j,kk,ii,l,jj,js;
  7.     double *e,*a,*b;
  8.     e=malloc(m*m*sizeof(double));
  9.     l=m;
  10.     if (l<n) l=n;
  11.     a=malloc(l*l*sizeof(double));
  12.     b=malloc(l*l*sizeof(double));
  13.     for (i=0; i<=n-1; i++)
  14.       for (j=0; j<=n-1; j++)
  15.         { ii=i*l+j; a[ii]=0.0;
  16.           for (kk=0; kk<=n-1; kk++)
  17.             a[ii]=a[ii]+p[i*n+kk]*f[j*n+kk];
  18.         }
  19.     for (i=0; i<=n-1; i++)
  20.       for (j=0; j<=n-1; j++)
  21.         { ii=i*n+j; p[ii]=q[ii];
  22.           for (kk=0; kk<=n-1; kk++)
  23.             p[ii]=p[ii]+f[i*n+kk]*a[kk*l+j];
  24.         }
  25.     for (ii=2; ii<=k; ii++)
  26.       { for (i=0; i<=n-1; i++)
  27.         for (j=0; j<=m-1; j++)
  28.           { jj=i*l+j; a[jj]=0.0;
  29.             for (kk=0; kk<=n-1; kk++)
  30.               a[jj]=a[jj]+p[i*n+kk]*h[j*n+kk];
  31.           }
  32.         for (i=0; i<=m-1; i++)
  33.         for (j=0; j<=m-1; j++)
  34.           { jj=i*m+j; e[jj]=r[jj];
  35.             for (kk=0; kk<=n-1; kk++)
  36.               e[jj]=e[jj]+h[i*n+kk]*a[kk*l+j];
  37.           }
  38.         js=rinv(e,m);
  39.         if (js==0) 
  40.           { free(e); free(a); free(b); return(js);}
  41.         for (i=0; i<=n-1; i++)
  42.         for (j=0; j<=m-1; j++)
  43.           { jj=i*m+j; g[jj]=0.0;
  44.             for (kk=0; kk<=m-1; kk++)
  45.               g[jj]=g[jj]+a[i*l+kk]*e[j*m+kk];
  46.           }
  47.         for (i=0; i<=n-1; i++)
  48.           { jj=(ii-1)*n+i; x[jj]=0.0;
  49.             for (j=0; j<=n-1; j++)
  50.               x[jj]=x[jj]+f[i*n+j]*x[(ii-2)*n+j];
  51.           }
  52.         for (i=0; i<=m-1; i++)
  53.           { jj=i*l; b[jj]=y[(ii-1)*m+i];
  54.             for (j=0; j<=n-1; j++)
  55.               b[jj]=b[jj]-h[i*n+j]*x[(ii-1)*n+j];
  56.           }
  57.         for (i=0; i<=n-1; i++)
  58.           { jj=(ii-1)*n+i;
  59.             for (j=0; j<=m-1; j++)
  60.               x[jj]=x[jj]+g[i*m+j]*b[j*l];
  61.           }
  62.         if (ii<k)
  63.           { for (i=0; i<=n-1; i++)
  64.             for (j=0; j<=n-1; j++)
  65.               { jj=i*l+j; a[jj]=0.0;
  66.                 for (kk=0; kk<=m-1; kk++)
  67.                   a[jj]=a[jj]-g[i*m+kk]*h[kk*n+j];
  68.                 if (i==j) a[jj]=1.0+a[jj];
  69.               }
  70.             for (i=0; i<=n-1; i++)
  71.             for (j=0; j<=n-1; j++)
  72.               { jj=i*l+j; b[jj]=0.0;
  73.                 for (kk=0; kk<=n-1; kk++)
  74.                   b[jj]=b[jj]+a[i*l+kk]*p[kk*n+j];
  75.               }
  76.             for (i=0; i<=n-1; i++)
  77.             for (j=0; j<=n-1; j++)
  78.               { jj=i*l+j; a[jj]=0.0;
  79.                 for (kk=0; kk<=n-1; kk++)
  80.                   a[jj]=a[jj]+b[i*l+kk]*f[j*n+kk];
  81.               }
  82.             for (i=0; i<=n-1; i++)
  83.             for (j=0; j<=n-1; j++)
  84.               { jj=i*n+j; p[jj]=q[jj];
  85.                 for (kk=0; kk<=n-1; kk++)
  86.                   p[jj]=p[jj]+f[i*n+kk]*a[j*l+kk];
  87.               }
  88.           }
  89.       }
  90.     free(e); free(a); free(b);
  91.     return(js);
  92. }
复制代码
七、C++语言实现代码
kalman.h: interface for the kalman class.
  1. //
  2. //////////////////////////////////////////////////////////////////////
  3. #if !defined(AFX_KALMAN_H__ED3D740F_01D2_4616_8B74_8BF57636F2C0__INCLUDED_)
  4. #define AFX_KALMAN_H__ED3D740F_01D2_4616_8B74_8BF57636F2C0__INCLUDED_
  5. #if _MSC_VER > 1000
  6. #pragma once
  7. #endif // _MSC_VER > 1000
  8. #include <math.h>
  9. #include "cv.h"
  10. class kalman 
  11. {
  12. public:
  13. void init_kalman(int x,int xv,int y,int yv);
  14. CvKalman* cvkalman;
  15. CvMat* state; 
  16. CvMat* process_noise;
  17. CvMat* measurement;
  18. const CvMat* prediction;
  19. CvPoint2D32f get_predict(float x, float y);
  20. kalman(int x=0,int xv=0,int y=0,int yv=0);
  21. //virtual ~kalman();

  22. };
  23. #endif // !defined(AFX_KALMAN_H__ED3D740F_01D2_4616_8B74_8BF57636F2C0__INCLUDED_)
复制代码
kalman.cpp
  1. #include "kalman.h"
  2. #include <stdio.h>

  3. /* tester de printer toutes les valeurs des vecteurs*/
  4. /* tester de changer les matrices du noises */
  5. /* replace state by cvkalman->state_post ??? */

  6. CvRandState rng;
  7. const double T = 0.1;
  8. kalman::kalman(int x,int xv,int y,int yv)
  9. {     
  10.     cvkalman = cvCreateKalman( 4, 4, 0 );
  11.     state = cvCreateMat( 4, 1, CV_32FC1 );
  12.     process_noise = cvCreateMat( 4, 1, CV_32FC1 );
  13.     measurement = cvCreateMat( 4, 1, CV_32FC1 );
  14.     int code = -1;
  15.     
  16.     /* create matrix data */
  17.      const float A[] = { 
  18.    1, T, 0, 0,
  19.    0, 1, 0, 0,
  20.    0, 0, 1, T,
  21.    0, 0, 0, 1
  22. };
  23.      
  24.      const float H[] = { 
  25.     1, 0, 0, 0,
  26.     0, 0, 0, 0,
  27.    0, 0, 1, 0,
  28.    0, 0, 0, 0
  29. };
  30.        
  31.      const float P[] = {
  32.     pow(320,2), pow(320,2)/T, 0, 0,
  33.    pow(320,2)/T, pow(320,2)/pow(T,2), 0, 0,
  34.    0, 0, pow(240,2), pow(240,2)/T,
  35.    0, 0, pow(240,2)/T, pow(240,2)/pow(T,2)
  36.     };
  37.      const float Q[] = {
  38.    pow(T,3)/3, pow(T,2)/2, 0, 0,
  39.    pow(T,2)/2, T, 0, 0,
  40.    0, 0, pow(T,3)/3, pow(T,2)/2,
  41.    0, 0, pow(T,2)/2, T
  42.    };
  43.    
  44.      const float R[] = {
  45.    1, 0, 0, 0,
  46.    0, 0, 0, 0,
  47.    0, 0, 1, 0,
  48.    0, 0, 0, 0
  49.    };
  50.    
  51.     
  52.     cvRandInit( &rng, 0, 1, -1, CV_RAND_UNI );
  53.     cvZero( measurement );
  54.     
  55.     cvRandSetRange( &rng, 0, 0.1, 0 );
  56.     rng.disttype = CV_RAND_NORMAL;
  57.     cvRand( &rng, state );
  58.     memcpy( cvkalman->transition_matrix->data.fl, A, sizeof(A));
  59.     memcpy( cvkalman->measurement_matrix->data.fl, H, sizeof(H));
  60.     memcpy( cvkalman->process_noise_cov->data.fl, Q, sizeof(Q));
  61.     memcpy( cvkalman->error_cov_post->data.fl, P, sizeof(P));
  62.     memcpy( cvkalman->measurement_noise_cov->data.fl, R, sizeof(R));
  63.     //cvSetIdentity( cvkalman->process_noise_cov, cvRealScalar(1e-5) );    
  64.     //cvSetIdentity( cvkalman->error_cov_post, cvRealScalar(1));
  65. //cvSetIdentity( cvkalman->measurement_noise_cov, cvRealScalar(1e-1) );
  66.     /* choose initial state */
  67.     state->data.fl[0]=x;
  68.     state->data.fl[1]=xv;
  69.     state->data.fl[2]=y;
  70.     state->data.fl[3]=yv;
  71.     cvkalman->state_post->data.fl[0]=x;
  72.     cvkalman->state_post->data.fl[1]=xv;
  73.     cvkalman->state_post->data.fl[2]=y;
  74.     cvkalman->state_post->data.fl[3]=yv;
  75. cvRandSetRange( &rng, 0, sqrt(cvkalman->process_noise_cov->data.fl[0]), 0 );
  76.     cvRand( &rng, process_noise );

  77.     }
  78.      
  79. CvPoint2D32f kalman::get_predict(float x, float y){
  80.    
  81.     /* update state with current position */
  82.     state->data.fl[0]=x;
  83.     state->data.fl[2]=y;
  84.     
  85.     /* predict point position */
  86.     /* x'k=A鈥 k+B鈥 k
  87.        P'k=A鈥 k-1*AT + Q */
  88.     cvRandSetRange( &rng, 0, sqrt(cvkalman->measurement_noise_cov->data.fl[0]), 0 );
  89.     cvRand( &rng, measurement );
  90.     
  91.      /* xk=A?xk-1+B?uk+wk */
  92.     cvMatMulAdd( cvkalman->transition_matrix, state, process_noise, cvkalman->state_post );
  93.     
  94.     /* zk=H?xk+vk */
  95.     cvMatMulAdd( cvkalman->measurement_matrix, cvkalman->state_post, measurement, measurement );
  96.     
  97.     /* adjust Kalman filter state */
  98.     /* Kk=P'k鈥 T鈥?H鈥 'k鈥 T+R)-1
  99.        xk=x'k+Kk鈥?zk-H鈥 'k)
  100.        Pk=(I-Kk鈥 )鈥 'k */
  101.     cvKalmanCorrect( cvkalman, measurement );
  102.     float measured_value_x = measurement->data.fl[0];
  103.     float measured_value_y = measurement->data.fl[2];
  104.     
  105. const CvMat* prediction = cvKalmanPredict( cvkalman, 0 );
  106.     float predict_value_x = prediction->data.fl[0];
  107.     float predict_value_y = prediction->data.fl[2];
  108.     return(cvPoint2D32f(predict_value_x,predict_value_y));
  109. }
  110. void kalman::init_kalman(int x,int xv,int y,int yv)
  111. {
  112. state->data.fl[0]=x;
  113.     state->data.fl[1]=xv;
  114.     state->data.fl[2]=y;
  115.     state->data.fl[3]=yv;
  116.     cvkalman->state_post->data.fl[0]=x;
  117.     cvkalman->state_post->data.fl[1]=xv;
  118.     cvkalman->state_post->data.fl[2]=y;
  119.     cvkalman->state_post->data.fl[3]=yv;
  120. }
3 0
原创粉丝点击