vis4d.zoo.base.models.qdtrack

QD-Track model config.

Functions

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

Get QDTrack model config.

get_qdtrack_yolox_cfg(num_classes, model_type)

Get QDTrack YOLOX model config.

get_qdtrack_cfg(num_classes, basemodel, weights=None)[source]

Get QDTrack model config.

Return type:

tuple[ConfigDict, ConfigDict]

get_qdtrack_yolox_cfg(num_classes, model_type, use_ema=True, weights=None)[source]

Get QDTrack YOLOX model config.

Return type:

tuple[ConfigDict, ConfigDict]