scorevision: push artifact

miner.py

@@ -24,6 +24,10 @@ from ultralytics import YOLO
 PERSON_CLS_ID = 0  # index into the element's objects list AND COCO's person id
 CONF_THRESHOLD = float(os.getenv("SV_CONF_THRESHOLD", "0.20"))
 IMG_SIZE = int(os.getenv("SV_IMG_SIZE", "416"))
+# Fine-tuned weights produce more duplicate boxes on the same person at the
+# ultralytics default (0.7); tightening NMS collapses most of them without
+# losing genuinely separate closely-spaced people.
+NMS_IOU = float(os.getenv("SV_NMS_IOU", "0.3"))
 
 
 class BoundingBox(BaseModel):
@@ -64,7 +68,7 @@ class Miner:
     def __repr__(self) -> str:
         return (
             f"Person detector: {self.model_name} "
-            f"(conf={CONF_THRESHOLD}, imgsz={IMG_SIZE})"
+            f"(conf={CONF_THRESHOLD}, imgsz={IMG_SIZE}, iou={NMS_IOU})"
         )
 
     def predict_batch(
@@ -79,6 +83,7 @@ class Miner:
             batch_images,
             conf=CONF_THRESHOLD,
             imgsz=IMG_SIZE,
+            iou=NMS_IOU,
             classes=[PERSON_CLS_ID],
             device=os.getenv("SV_DEVICE", "cpu"),
             verbose=False,