def refine_tracklets(self, event): DLCscorer, _ = auxiliaryfunctions.GetScorerName( self.cfg, self.shuffle.GetValue(), self.cfg["TrainingFraction"][-1], ) track_method = self.cfg.get("default_track_method", "ellipse") if track_method == "ellipse": method = "el" elif track_method == "box": method = "bx" else: method = "sk" dest = str(Path(self.video).parents[0]) vname = Path(self.video).stem datafile = os.path.join(dest, vname + DLCscorer + f"_{method}.h5") self.manager, self.viz = deeplabcut.refine_tracklets( self.config, datafile, self.video, min_swap_len=self.slider_swap.GetValue(), trail_len=self.length_track.GetValue(), max_gap=self.slider_gap.GetValue(), ) self.export.Enable()
def refine_tracklets(self, event): self.manager, self.viz = deeplabcut.refine_tracklets( self.config, self.datafile, self.video, max_gap=self.slider_gap.GetValue(), trail_len=self.length_track.GetValue(), ) self.export.Enable()