ParticleEx1.asv
上传用户:shigeng
上传日期:2017-01-30
资源大小:122k
文件大小:2k
开发平台:

Matlab

  1. function Particle
  2. % Particle filter 
  3. x = 0.1; % 初始状态
  4. Q = 1; % 过程噪声协方差
  5. R = 1; % 测量噪声协方差
  6. tf = 50; % 仿真长度
  7. N = 100; % 粒子滤波器粒子数
  8. xhat = x;
  9. P = 2;
  10. xhatPart = x;
  11. % 初始化粒子过滤器
  12. for i = 1 : N
  13.     xpart(i) = x + sqrt(P) * randn;
  14. end
  15. xArr = [x];
  16. yArr = [x^2 / 20 + sqrt(R) * randn];
  17. xhatArr = [x];
  18. PArr = [P];
  19. xhatPartArr = [xhatPart];
  20. close all;
  21. for k = 1 : tf
  22.     % 系统仿真
  23.     x = 0.5 * x + 25 * x / (1 + x^2) + 8 * cos(1.2*(k-1)) + sqrt(Q) * randn;%状态方程
  24.     y = x^2 / 20 + sqrt(R) * randn;%观测方程
  25.    
  26.     for i = 1 : N
  27.         xpartminus(i) = 0.5 * xpart(i) + 25 * xpart(i) / (1 + xpart(i)^2) + 8 * cos(1.2*(k-1)) + sqrt(Q) * randn;
  28.         ypart = xpartminus(i)^2 / 20;
  29.         vhat = y - ypart;%观测和预测的差
  30.         q(i) = (1 / sqrt(R) / sqrt(2*pi)) * exp(-vhat^2 / 2 / R);
  31.     end
  32.     %正常化的可能性,每个先验估计
  33.     qsum = sum(q);
  34.     for i = 1 : N
  35.         q(i) = q(i) / qsum;%归一化权重
  36.     end
  37.     % 重采样
  38.     for i = 1 : N
  39.         u = rand; % 均匀随机数介于0和1
  40.         qtempsum = 0;
  41.         for j = 1 : N
  42.             qtempsum = qtempsum + q(j);
  43.             if qtempsum >= u
  44.                 xpart(i) = xpartminus(j);
  45.                 break;
  46.             end
  47.         end
  48.     end
  49.     
  50.     xhatPart = mean(xpart);
  51.     % Plot the estimated pdf's at a specific time.
  52.     if k == 20
  53.         % Particle filter pdf
  54.         pdf = zeros(81,1);
  55.         for m = -40 : 40
  56.             for i = 1 : N
  57.                 if (m <= xpart(i)) && (xpart(i) < m+1)
  58.                     pdf(m+41) = pdf(m+41) + 1;
  59.                 end
  60.             end
  61.         end
  62.         
  63.     xhatArr = [xhatArr xhat];
  64.     PArr = [PArr P];
  65.     xhatPartArr = [xhatPartArr xhatPart];
  66. end
  67. t = 0 : tf;
  68. figure;
  69. plot(t, xArr, 'b.', t, xhatPartArr, 'k-');
  70. set(gca,'FontSize',12); set(gcf,'Color','White'); 
  71. xlabel('time step'); ylabel('state');
  72. legend('True state', 'Particle filter estimate');