Upload folder using huggingface_hub

miner.py

@@ -181,7 +181,7 @@ class Miner:
         boxes: np.ndarray,
         scores: np.ndarray,
         sigma: float = 0.5,
-        score_thresh: float = 0.01,
+        score_thresh: float = 0.2,
     ) -> tuple[np.ndarray, np.ndarray]:
         """
         Soft-NMS: Gaussian decay of overlapping scores instead of hard removal.
@@ -264,7 +264,7 @@ class Miner:
         boxes = self._clip_boxes(boxes, (orig_w, orig_h))
 
         if apply_optional_dedup and len(boxes) > 1:
-            keep_idx, scores = self._soft_nms(boxes, scores)
+            keep_idx, scores = self._soft_nms(boxes, scores, sigma=self.iou_thres, score_thresh=self.conf_thres)
             boxes = boxes[keep_idx]
             cls_ids = cls_ids[keep_idx]