parent
							
								
									4917e22615
								
							
						
					
					
						commit
						e7418cfa39
					
				
				 1 changed files with 10 additions and 0 deletions
			
			
		@ -1,3 +1,13 @@ | 
				
			||||
# pylint: skip-file | 
				
			||||
from openpilot.common.kalman.simple_kalman_impl import KF1D as KF1D | 
				
			||||
assert KF1D | 
				
			||||
import numpy as np | 
				
			||||
 | 
				
			||||
def get_kalman_gain(dt, A, C, Q, R, iterations=100): | 
				
			||||
  P = np.zeros_like(Q) | 
				
			||||
  for _ in range(iterations): | 
				
			||||
    P = A.dot(P).dot(A.T) + dt * Q | 
				
			||||
    S = C.dot(P).dot(C.T) + R | 
				
			||||
    K = P.dot(C.T).dot(np.linalg.inv(S)) | 
				
			||||
    P = (np.eye(len(P)) - K.dot(C)).dot(P) | 
				
			||||
  return K | 
				
			||||
					Loading…
					
					
				
		Reference in new issue