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

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.         figure;
  63.         m = -40 : 40;
  64.         plot(m, pdf / N, 'r');
  65.         hold;
  66.         title('Estimated pdf at k=20');
  67.         disp(['min, max xpart(i) at k = 20: ', num2str(min(xpart)), ', ', num2str(max(xpart))]);
  68.         % Kalman filter pdf
  69.         pdf = (1 / sqrt(P) / sqrt(2*pi)) .* exp(-(m - xhat).^2 / 2 / P);
  70.         plot(m, pdf, 'b');
  71.         legend('Particle filter', 'Kalman filter');
  72.     end
  73.     % Save data in arrays for later plotting
  74.     xArr = [xArr x];
  75.     yArr = [yArr y];
  76.     xhatArr = [xhatArr xhat];
  77.     PArr = [PArr P];
  78.     xhatPartArr = [xhatPartArr xhatPart];
  79. end
  80. t = 0 : tf;
  81. %figure;
  82. %plot(t, xArr);
  83. %ylabel('true state');
  84. figure;
  85. plot(t, xArr, 'b.', t, xhatPartArr, 'k-');
  86. set(gca,'FontSize',12); set(gcf,'Color','White'); 
  87. xlabel('time step'); ylabel('state');
  88. legend('True state', 'Particle filter estimate'); 
  89. xhatRMS = sqrt((norm(xArr - xhatArr))^2 / tf);
  90. xhatPartRMS = sqrt((norm(xArr - xhatPartArr))^2 / tf);
  91. disp(['Kalman filter RMS error = ', num2str(xhatRMS)]);
  92. disp(['Particle filter RMS error = ', num2str(xhatPartRMS)]);