有没有cv2.KalmanFilter实现的例子？

Question

我正在尝试使用Python包装的OpenCV (cv2) 来构建一个非常简单的用于2D对象追踪的跟踪器。

我只注意到了3个函数:

• KalmanFilter（构造函数）
• .predict()
• .correct(measurement)

我的想法是创建一段代码来检查Kalman是否正常工作，如下所示:

kf = cv2.KalmanFilter(...)
# set initial position

cv2.predict()
corrected_position = cv2.correct([measurement_x, measurement_y])

我找到了一些使用cv包装器的例子，但没有找到使用cv2的...

先行感谢！

- Jairo Vadillo

Answer 1

如果你正在使用的是OpenCV2.4，那就不太好了：KalmanFilter是无法使用的，因为你无法设置状态转移（或任何其他）矩阵。

对于OpenCV3.0，它会正常工作，像这样：

import cv2, numpy as np

meas=[]
pred=[]
frame = np.zeros((400,400,3), np.uint8) # drawing canvas
mp = np.array((2,1), np.float32) # measurement
tp = np.zeros((2,1), np.float32) # tracked / prediction

def onmouse(k,x,y,s,p):
    global mp,meas
    mp = np.array([[np.float32(x)],[np.float32(y)]])
    meas.append((x,y))

def paint():
    global frame,meas,pred
    for i in range(len(meas)-1): cv2.line(frame,meas[i],meas[i+1],(0,100,0))
    for i in range(len(pred)-1): cv2.line(frame,pred[i],pred[i+1],(0,0,200))

def reset():
    global meas,pred,frame
    meas=[]
    pred=[]
    frame = np.zeros((400,400,3), np.uint8)

cv2.namedWindow("kalman")
cv2.setMouseCallback("kalman",onmouse);
kalman = cv2.KalmanFilter(4,2)
kalman.measurementMatrix = np.array([[1,0,0,0],[0,1,0,0]],np.float32)
kalman.transitionMatrix = np.array([[1,0,1,0],[0,1,0,1],[0,0,1,0],[0,0,0,1]],np.float32)
kalman.processNoiseCov = np.array([[1,0,0,0],[0,1,0,0],[0,0,1,0],[0,0,0,1]],np.float32) * 0.03
#kalman.measurementNoiseCov = np.array([[1,0],[0,1]],np.float32) * 0.00003
while True:
    kalman.correct(mp)
    tp = kalman.predict()
    pred.append((int(tp[0]),int(tp[1])))
    paint()
    cv2.imshow("kalman",frame)
    k = cv2.waitKey(30) &0xFF
    if k == 27: break
    if k == 32: reset()

- berak

---

谢谢提供信息，不幸的是需要使用opencv 2.49... 我们正在尝试使用pykalman :) - Jairo Vadillo

比其他代码简单得多。但是有一个问题，最初我们必须先预测/估计然后再校正测量值，不是吗？ - Jumabek Alikhanov