fixes for tracking lost

Author: arch

funscript_editor/algorithms/funscriptgenerator.py

@@ -760,7 +760,7 @@ def tracking(self) -> str:
                 break
 
             if woman_tracker_status == StaticVideoTracker.Status.TRACKING_LOST:
-                if len(bboxes['Woman']) == 0 or max([x for x in bboxes['Woman'].keys()]) < tracking_lost_frames:
+                if len(bboxes['Woman']) == 0 or frame_num - max([x for x in bboxes['Woman'].keys()]) >= tracking_lost_frames:
                     status = 'Woman ' + woman_tracker_status
                     bboxes = self.delete_last_tracking_predictions(bboxes, (self.params.skip_frames+1)*2)
                     break
@@ -773,7 +773,7 @@ def tracking(self) -> str:
                     break
 
                 if men_tracker_status == StaticVideoTracker.Status.TRACKING_LOST:
-                    if len(bboxes['Men']) == 0 or max([x for x in bboxes['Men'].keys()]) < tracking_lost_frames:
+                    if len(bboxes['Men']) == 0 or frame_num -  max([x for x in bboxes['Men'].keys()]) >= tracking_lost_frames:
                         status = 'Men ' + men_tracker_status
                         bboxes = self.delete_last_tracking_predictions(bboxes, (self.params.skip_frames+1)*2)
                         break