vis4d.op.motion.velo_lstm

VeloLSTM operations.

Classes

VeloLSTMLoss([loc_dim, smooth_weight])

Loss term for VeloLSTM.

class VeloLSTMLoss(loc_dim=7, smooth_weight=0.001)[source]

Loss term for VeloLSTM.

__init__(loc_dim=7, smooth_weight=0.001)[source]: Initialize the loss term.

static linear_motion_loss(outputs)[source]

Linear motion loss.

Loss: |(loc_t - loc_t-1), (loc_t-1, loc_t-2)|_1 for t = [2, s_len]

forward(loc_preds, loc_refines, gt_traj)[source]

Loss term for VeloLSTM.