tracklab.wrappers.tracklet_agg package

Submodules

tracklab.wrappers.tracklet_agg.majority_vote_api module

class tracklab.wrappers.tracklet_agg.majority_vote_api.MajorityVoteTracklet(cfg, device, tracking_dataset=None)[source]

Bases: VideoLevelModule

input_columns = []
output_columns = []
process(detections: DataFrame, metadatas: DataFrame)[source]

The main processing function. Runs on GPU.

Parameters:
  • detections – The previous detections.

  • metadatas – The previous image metadatas

Returns:

Either a DataFrame containing the new/updated detections

or a tuple containing detections and metadatas (in that order) The DataFrames can be either a list of Series, a list of DataFrames or a single DataFrame. The returned objects will be aggregated automatically according to the name of the Series/index of the DataFrame. It is thus mandatory here to name correctly your series or index your dataframes. The output will override the previous detections with the same name/index.

Return type:

output