vis4d.state.track3d.motion

3D Motional Models.

class BaseMotionModel(num_frames, motion_dims, hits=1, hit_streak=0, time_since_update=0, age=0, fps=1)

Base class for motion model.

__init__(num_frames, motion_dims, hits=1, hit_streak=0, time_since_update=0, age=0, fps=1)

Creates an instance of the class.

update(obs_3d, info)

Update the state.

Return type:

None

predict_velocity()

Predict velocity.

Return type:

Tensor

predict(update_state=True)

Predict the state.

Return type:

Tensor

get_state()

Get the state.

Return type:

Tensor

class KF3DMotionModel(*args, obs_3d, **kwargs)

Kalman filter 3D motion model.

__init__(*args, obs_3d, **kwargs)

Creates an instance of the class.

update(obs_3d, info)

Update the state.

Return type:

None

predict_velocity()

Predict velocity.

Return type:

Tensor

predict(update_state=True)

Predict the state.

Return type:

Tensor

get_state()

Returns the current bounding box estimate.

Return type:

Tensor

class LSTM3DMotionModel(*args, lstm_model, obs_3d, init_flag=True, **kwargs)

LSTM 3D motion model.

__init__(*args, lstm_model, obs_3d, init_flag=True, **kwargs)

Initialize a motion model using initial bounding box.

update(obs_3d, info)

Updates the state vector with observed bbox.

Return type:

None

predict_velocity()

Predict velocity.

Return type:

Tensor

predict(update_state=True)

Advances the state vector and returns the predicted bounding box.

Return type:

Tensor

get_state()

Returns the current bounding box estimate.

Return type:

Tensor

Modules

base	Motion model base class.
kf3d	Kalman Filter 3D motion model.
lstm_3d	LSTM 3D motion model.