sportslabkit.matching.base_batch#

assignment cost calculation & matching methods.

Overview#

Classes#

BaseBatchMatchingFunction

A base class for batch matching functions.
Attributes#

EPS

-

Node

-

Classes#

class sportslabkit.matching.base_batch.BaseBatchMatchingFunction[source]#

A base class for batch matching functions.

A batch matching function takes a list of trackers and a list of detections and returns a list of matches.

Overview

Methods#

compute_cost_matrices(trackers, list_of_detections)

abc Calculate the cost matrix between trackers and detections.

Members

abstract compute_cost_matrices(trackers: list[sportslabkit.Tracklet], list_of_detections: list[list[sportslabkit.types.detection.Detection]]) → list[numpy.ndarray][source]#

Calculate the cost matrix between trackers and detections.

Parameters:

  • trackers – A list of trackers.

  • list_of_detections – A list containing a list of detections for each frame.

Returns:

A list of 2D numpy arrays where the element at [i, j] in the kth array is the cost between tracker i and detection j in frame k.

Attributes#

sportslabkit.matching.base_batch.EPS = 1e-07[source]#
sportslabkit.matching.base_batch.Node[source]#