YOLO

operators/generate_mask.py

@@ -21,21 +21,21 @@ class SEQUENCER_OT_generate_face_mask(Operator):
     bl_description = "Detect faces and generate mask image sequence"
     bl_options = {'REGISTER', 'UNDO'}
     
-    # Detection parameters
-    scale_factor: FloatProperty(
-        name="Scale Factor",
-        description="Detection scale factor (larger = faster but less accurate)",
-        default=1.1,
-        min=1.01,
-        max=2.0,
+    # YOLO Detection parameters
+    conf_threshold: FloatProperty(
+        name="Confidence",
+        description="YOLO confidence threshold (higher = fewer false positives)",
+        default=0.25,
+        min=0.1,
+        max=1.0,
     )
     
-    min_neighbors: IntProperty(
-        name="Min Neighbors",
-        description="Minimum neighbors for detection (higher = fewer false positives)",
-        default=5,
-        min=1,
-        max=20,
+    iou_threshold: FloatProperty(
+        name="IOU Threshold",
+        description="Non-maximum suppression IOU threshold",
+        default=0.45,
+        min=0.1,
+        max=1.0,
     )
     
     mask_scale: FloatProperty(
@@ -133,8 +133,8 @@ class SEQUENCER_OT_generate_face_mask(Operator):
             start_frame=0,  # Frame indices in video
             end_frame=end_frame - start_frame,
             fps=fps,
-            scale_factor=self.scale_factor,
-            min_neighbors=self.min_neighbors,
+            conf_threshold=self.conf_threshold,
+            iou_threshold=self.iou_threshold,
             mask_scale=self.mask_scale,
             on_complete=on_complete,
             on_progress=on_progress,