Upload folder using huggingface_hub
Browse files
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 / "
|
| 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.
|
| 76 |
|
| 77 |
# High-confidence boxes can survive without TTA confirmation
|
| 78 |
-
self.conf_high = 0.
|
| 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
|