SuperBitDev commited on
Commit
cf5028c
·
verified ·
1 Parent(s): 81e0213

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. miner.py +3 -3
miner.py CHANGED
@@ -25,7 +25,7 @@ class TVFrameResult(BaseModel):
25
 
26
  class Miner:
27
  def __init__(self, path_hf_repo: Path) -> None:
28
- model_path = path_hf_repo / "person-detection.onnx"
29
  self.class_names = ["person"]
30
  print("ORT version:", ort.__version__)
31
 
@@ -72,10 +72,10 @@ class Miner:
72
 
73
  # ---------- Scoring-oriented thresholds ----------
74
  # Low threshold for candidate generation
75
- self.conf_thres = 0.10
76
 
77
  # High-confidence boxes can survive without TTA confirmation
78
- self.conf_high = 0.20
79
 
80
  # NMS threshold
81
  self.iou_thres = 0.50
 
25
 
26
  class Miner:
27
  def __init__(self, path_hf_repo: Path) -> None:
28
+ model_path = path_hf_repo / "weights.onnx"
29
  self.class_names = ["person"]
30
  print("ORT version:", ort.__version__)
31
 
 
72
 
73
  # ---------- Scoring-oriented thresholds ----------
74
  # Low threshold for candidate generation
75
+ self.conf_thres = 0.05
76
 
77
  # High-confidence boxes can survive without TTA confirmation
78
+ self.conf_high = 0.25
79
 
80
  # NMS threshold
81
  self.iou_thres = 0.50