예제 #1
0
 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()
예제 #2
0
 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()