vis4d.op.track.matching

Matching calculation utils.

Functions

calc_bisoftmax_affinity(...[, ...])	Calculate affinity matrix using bisoftmax metric.
cosine_similarity(key_embeds, ref_embeds[, ...])	Calculate cosine similarity.

calc_bisoftmax_affinity(detection_embeddings, track_embeddings, detection_class_ids=None, track_class_ids=None, with_categories=False)

Calculate affinity matrix using bisoftmax metric.

Return type:

Tensor

cosine_similarity(key_embeds, ref_embeds, normalize=True, temperature=-1)

Calculate cosine similarity.

Return type:

Tensor