Source code for ddd_pylib.tracking._laptrack_strategy

from ddd_pylib.tracking._base_tracking_strategy import BaseTrackingStrategy
from laptrack import LapTrack
import pandas as pd




class LapTrackStrategy(BaseTrackingStrategy):
    def __init__(self):
        super().__init__()

    def _linkTracks(self):
        d = self.getDetections()
        coords_cols = [a for a in ['Z', 'Y', 'X'] if a in d.columns]
        lt = LapTrack(cutoff=15**2, splitting_cutoff=30**2, gap_closing_max_frame_count=self.getMemory())
        track_df, _, _ = lt.predict_dataframe(d, coordinate_cols=coords_cols, only_coordinate_cols=False, frame_col="T")
        track_df["track_id"] += 1
        track_df = track_df.reset_index()
        return track_df