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