Update miner.py

miner.py

@@ -24,19 +24,7 @@ class TVFrameResult(BaseModel):
 
 
 class Miner:
-    def __init__(self,
-        path_hf_repo: Path,
-        conf_thres: float = 0.21,
-        conf_high: float = 0.22,
-        iou_thres: float = 0.77,
-        tta_match_iou: float = 0.56,
-        max_det: int = 264,
-        min_box_area: int = 82,
-        min_w: int = 15,
-        min_h: int = 17,
-        max_aspect_ratio: float = 3.65,
-        max_box_area_ratio: float = 0.78,
-    ) -> None:
+    def __init__(self, path_hf_repo: Path) -> None:
         model_path = path_hf_repo / "weights.onnx"
 
         self.class_names = ['bus', 'car', 'truck', 'motorcycle']
@@ -90,26 +78,26 @@ class Miner:
 
         # ---------- Scoring-oriented thresholds ----------
         # Low threshold for candidate generation
-        self.conf_thres = conf_thres
+        self.conf_thres = 0.2124
 
         # High-confidence boxes can survive without TTA confirmation
-        self.conf_high = conf_high
+        self.conf_high = 0.7225
 
         # NMS threshold
-        self.iou_thres = iou_thres
+        self.iou_thres = 0.7704
 
         # TTA confirmation IoU
-        self.tta_match_iou = tta_match_iou
+        self.tta_match_iou = 0.5609
 
-        self.max_det = max_det
+        self.max_det = 264
         self.use_tta = True
 
         # Box sanity filters
-        self.min_box_area = min_box_area
-        self.min_w = min_w
-        self.min_h = min_h
-        self.max_aspect_ratio = max_aspect_ratio
-        self.max_box_area_ratio = max_box_area_ratio
+        self.min_box_area = 82
+        self.min_w = 15
+        self.min_h = 17
+        self.max_aspect_ratio = 3.6571
+        self.max_box_area_ratio = 0.7807
 
         print(f"✅ ONNX model loaded from: {model_path}")
         print(f"✅ ONNX providers: {self.session.get_providers()}")