85 lines
2.6 KiB
Python
85 lines
2.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
Created on Fri Oct 18 18:02:07 2013
|
|
|
|
@author: rlabbe
|
|
"""
|
|
|
|
import numpy as np
|
|
import scipy.linalg as linalg
|
|
import matplotlib.pyplot as plt
|
|
import numpy.random as random
|
|
|
|
class KalmanFilter:
|
|
|
|
def __init__(self, dim):
|
|
""" Create a Kalman filter of dimension 'dim'"""
|
|
|
|
self.x = 0 # state
|
|
self.P = np.matrix(np.eye(dim)) # uncertainty covariance
|
|
self.Q = np.matrix(np.eye(dim)) # process uncertainty
|
|
self.u = np.matrix(np.zeros((dim,1))) # motion vector
|
|
self.B = 0
|
|
self.F = 0 # state transition matrix
|
|
self.H = 0 # Measurement function (maps state to measurements)
|
|
self.R = np.matrix(np.eye(1)) # state uncertainty
|
|
self.I = np.matrix(np.eye(dim))
|
|
|
|
|
|
def update(self, Z):
|
|
"""
|
|
Add a new measurement to the kalman filter.
|
|
"""
|
|
|
|
# measurement update
|
|
y = Z - (self.H * self.x) # error (residual) between measurement and prediction
|
|
S = (self.H * self.P * self.H.T) + self.R # project system uncertainty into measurment space + measurement noise(R)
|
|
|
|
|
|
K = self.P * self.H.T * linalg.inv(S) # map system uncertainty into kalman gain
|
|
|
|
self.x = self.x + (K*y) # predict new x with residual scaled by the kalman gain
|
|
self.P = (self.I - (K*self.H))*self.P # and compute the new covariance
|
|
|
|
def predict(self):
|
|
# prediction
|
|
self.x = (self.F*self.x) + self.u
|
|
self.P = self.F * self.P * self.F.T + self.Q
|
|
|
|
|
|
if __name__ == "__main__":
|
|
f = KalmanFilter (dim=2)
|
|
|
|
f.x = np.matrix([[2.],
|
|
[0.]]) # initial state (location and velocity)
|
|
|
|
f.F = np.matrix([[1.,1.],
|
|
[0.,1.]]) # state transition matrix
|
|
|
|
f.H = np.matrix([[1.,0.]]) # Measurement function
|
|
f.P *= 1000. # covariance matrix
|
|
f.R = 5 # state uncertainty
|
|
f.Q *= 0.0001 # process uncertainty
|
|
|
|
measurements = []
|
|
results = []
|
|
for t in range (100):
|
|
# create measurement = t plus white noise
|
|
z = t + random.randn()*20
|
|
|
|
# perform kalman filtering
|
|
f.measure (z)
|
|
f.predict()
|
|
|
|
# save data
|
|
results.append (f.x[0,0])
|
|
measurements.append(z)
|
|
|
|
# plot data
|
|
p1, = plt.plot(measurements,'r')
|
|
p2, = plt.plot (results,'b')
|
|
p3, = plt.plot ([0,100],[0,100], 'g') # perfect result
|
|
plt.legend([p1,p2, p3], ["noisy measurement", "KF output", "ideal"], 4)
|
|
|
|
|
|
plt.show() |