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.

Initialize the loss term.

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

Loss term for VeloLSTM.

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]