vis4d.zoo.cc_3dt.model

CC-3DT model config.

Functions

get_cc_3dt_cfg(num_classes, basemodel[, ...])

Get CC-3DT model config.
get_cc_3dt_cfg(num_classes, basemodel, pure_det=False, motion_model='KF3D', lstm_model=None, fps=2)[source]

Get CC-3DT model config.

Parameters:

  • num_classes (int) – Number of classes.

  • basemodel (ConfigDict) – Base model config.

  • pure_det (bool, optional) – Whether to use pure detection mode. Defaults to False.

  • motion_model (str, optional) – Motion model. Defaults to “KF3D”.

  • lstm_model (ConfigDict, optional) – LSTM model config. Defaults to None.

  • fps (int, optional) – FPS. Defaults to 2.

Return type:

tuple[ConfigDict, ConfigDict]