KF与EKF代码demo(几分钟就能让人人都会使用)

本文中代码均为C#实现,可自行换为其他语言。

1.KF,对数据自身或者有对应观测值的数据进行滤波(一维线性)

namespace ZaneCon.PredictedState.FeederState
{
    //本次卡尔曼针对的状态量与观测值均为特定情境下一维线性
    public class FeederKF
    {
        private double A; // 状态转移矩阵
        private double H; // 观测矩阵
        private double Q; // 状态噪声协方差矩阵
        private double R; // 观测噪声协方差矩阵
        private double P; // 状态估计协方差矩阵
        private double x; // 状态估计向量
        public FeederKF()
        {
            // 初始化矩阵和向量
  //因为一维线性,比如模型为y=2x或直接对数据过滤(没有线性关系,一组数据或两组数据——一组状态量+一组观测)
            A = 1;
            H = 1;
            Q = 0.1;
            R = 1;
            P = 1;
            x = 0;
        }
       //这里x为状态量,z为对应的观测数据
        public double Filter(double x, double z)
        {
            // 预测步骤
            double x_ = A * x;
            double P_ = A * P * A + Q;
            // 更新步骤
            double K = P_ * H / (H * P_ * H + R);
            x = x_ + K * (z - H * x_);
            P = (1 - K * H) * P_;
            return x;
        }
    }
}

上述例子应用对象为两组数据进行融合滤波,比如测得A组数据为一系列值——数组A,A组观测数据为一系列值——数组B,现融合A和B得到更准的A数据。

2.EKF,与KF相比就是A阵不同,一般对非线性模型进行线性化,其他步骤一模一样。

namespace ZaneCon.PredictedState.FeederState
{
    //EKF完整例子,用时修改状态量、量测量、维数及A阵即可
    public class FeederEKF
    {
        // 状态转移矩阵
        private Matrix<double> A; //状态转移矩阵
        private Matrix<double> H; // 观测矩阵
        private Matrix<double> Q; // 状态噪声协方差矩阵(肯定与状态量维数一样)
        private Matrix<double> R; // 观测噪声协方差矩阵(肯定与观测量维数一样)
        private Matrix<double> P; // 状态估计协方差矩阵
        private Matrix<double> X; // 状态估计向量6维,量测2维

        public FeederEKF()
        {
            // 初始化矩阵和向量
            A = DenseMatrix.CreateIdentity(6);//6*6的单位阵,状态维度*状态维度
            H = Matrix<double>.Build.DenseOfArray(new double[,] { { 1, 0, 0, 0, 0, 0 }, { 0, 1, 0, 0, 0, 0 } });//2*6,量测维度*状态维度
            Q = DenseMatrix.CreateIdentity(6);//6*6状态噪声协方差矩阵,状态维度*状态维度
            R = Matrix<double>.Build.DenseOfArray(new double[,] { { 1, 0 }, { 0, 1 } });//2*2观测噪声协方差矩阵,量测维度*量测维度
            // 状态估计协方差矩阵,状态维度*状态维度
            P = Matrix<double>.Build.DenseOfArray(new double[,] { { 10, 0, 0, 0, 0, 0 }, { 0, 10, 0, 0, 0, 0 }, { 0, 0, 10, 0, 0, 0 }, { 0, 0, 0, 10, 0, 0 }, { 0, 0, 0, 0, 10, 0 }, { 0, 0, 0, 0, 0, 10 } });
            X = Matrix<double>.Build.Dense(6, 1, 0.0); ;//6*1
        }
      //此处状态量为x、y、thta、v、w、a,观测量为z_x、z_y
        public Matrix<double> Filter(double x, double y, double thta, double v, double w, double a, double delta_t, double z_x, double z_y)
        {
            /* 此处状态量及其机械编排(离散化)为:x=x+v*Cos(thta);
                         y=y+v*Sin(thta);
                         thta=thta+wt;
                         v=v+at;
                         w
                         a
               量测为x,y;
             */
            A[0, 2] = -v * Asin(thta);//thta必须在-1~1之间,不然出现NAN
            A[0, 3] = Acos(thta);
            A[1, 2] = v * Acos(thta);
            A[1, 3] = Asin(thta);
            A[3, 4] = delta_t;
            A[4, 5] = delta_t;
            Matrix<double> Z = Matrix<double>.Build.DenseOfArray(new double[,] { { z_x }, { z_y } });//量测阵
            X = Matrix<double>.Build.DenseOfArray(new double[,] { { x }, { y }, { thta }, { v }, { w }, { a } });
            // 预测步骤
            Matrix<double> X_ = A * X;
            Matrix<double> P_ = A * P * A.Transpose() + Q;
            // 更新步骤
            Matrix<double> K = (P_ * H.Transpose()) * (H * P_ * H.Transpose() + R).Inverse();
            X = X_ + K * (Z - H * X_);
            P = (1 - K * H) * P_;
            return X;
        }
    }
}

使用EKF一定要注意各个阵的维度设置,另外A阵雅可比的计算依赖于机械编排(模型式子求偏导),至于雅可比矩阵的排列可以不懂的可以百度一样,很简单。A阵求出来后,直接套用公式即可,与KF一样。
此处的例子是通过状态量间接算出测量值xy,然后与观测值xy进行融合滤波,得到更精确的xy。比如通过imu计算得到的xy与GPS得到的xy进行融合。

通过上述demo,现在是不是觉得晦涩难懂的理论在应用起来却是非常简单呢?

你可能感兴趣的:(自动驾驶,算法)