vis4d.state.track3d.motion.lstm_3d

LSTM 3D motion model.

Classes

LSTM3DMotionModel(*args, lstm_model, obs_3d)

LSTM 3D motion model.

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