use center point for supervised tracking

Author: arch

funscript_editor/algorithms/videotracker.py

@@ -100,13 +100,15 @@ def is_bbox_in_tracking_area(bbox: tuple, supervised_tracking_area: tuple = None
             return False
         if supervised_tracking_area[3] <= 1:
             return False
-        if bbox[0] < supervised_tracking_area[0]:
+
+        if bbox[4] < supervised_tracking_area[0]:
             return False
-        if bbox[1] < supervised_tracking_area[1]:
+        if bbox[5] < supervised_tracking_area[1]:
             return False
-        if bbox[0] + bbox[2] > supervised_tracking_area[0] + supervised_tracking_area[2]:
+
+        if bbox[4] > supervised_tracking_area[0] + supervised_tracking_area[2]:
             return False
-        if bbox[1] + bbox[3] > supervised_tracking_area[1] + supervised_tracking_area[3]:
+        if bbox[5] > supervised_tracking_area[1] + supervised_tracking_area[3]:
             return False
 
         return True

funscript_editor/ui/opencvui.py

@@ -213,7 +213,7 @@ def draw_box(self, bbox, color: tuple = (255, 0, 255)) -> None:
                     cv2.circle(
                             self.preview_image,
                             (box[4], box[5]),
-                            5,
+                            4,
                             color,
                             2
                         )
@@ -248,7 +248,7 @@ def draw_box_to_image(image: np.ndarray, bbox, color: tuple = (255, 0, 255)) ->
                     cv2.circle(
                             image,
                             (box[4], box[5]),
-                            5,
+                            4,
                             color,
                             2
                         )