vis4d.op.motion.kalman_filter

Kalman Filter PyTorch implementation.

Functions

predict(motion_mat, cov_motion_q, mean, ...)

Run Kalman filter prediction step.

project(update_mat, cov_project_r, mean, ...)

Project state distribution to measurement space.

update(update_mat, cov_project_r, mean, ...)

Run Kalman filter correction step.

predict(motion_mat, cov_motion_q, mean, covariance)[source]

Run Kalman filter prediction step.

Return type:

tuple[Tensor, Tensor]

project(update_mat, cov_project_r, mean, covariance)[source]

Project state distribution to measurement space.

Return type:

tuple[Tensor, Tensor]

update(update_mat, cov_project_r, mean, covariance, measurement)[source]

Run Kalman filter correction step.

Return type:

tuple[Tensor, Tensor]