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.

Creates an instance of the class.

get_state()

Get the state.

Return type:

Tensor

predict(update_state=True)

Predict the state.

Return type:

Tensor

predict_velocity()

Predict velocity.

Return type:

Tensor

update(obs_3d, info)

Update the state.

Return type:

None

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

Kalman filter 3D motion model.

Creates an instance of the class.

get_state()

Returns the current bounding box estimate.

Return type:

Tensor

predict(update_state=True)

Predict the state.

Return type:

Tensor

predict_velocity()

Predict velocity.

Return type:

Tensor

update(obs_3d, info)

Update the state.

Return type:

None

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

LSTM 3D motion model.

Initialize a motion model using initial bounding box.

get_state()

Returns the current bounding box estimate.

Return type:

Tensor

predict(update_state=True)

Advances the state vector and returns the predicted bounding box.

Return type:

Tensor

predict_velocity()

Predict velocity.

Return type:

Tensor

update(obs_3d, info)

Updates the state vector with observed bbox.

Return type:

None

Modules

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