Upload folder using huggingface_hub
Browse files
miner.py
CHANGED
|
@@ -72,7 +72,7 @@ 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.5
|
|
@@ -83,7 +83,7 @@ class Miner:
|
|
| 83 |
# TTA confirmation IoU
|
| 84 |
self.tta_match_iou = 0.55
|
| 85 |
|
| 86 |
-
self.max_det =
|
| 87 |
self.use_tta = True
|
| 88 |
|
| 89 |
# Box sanity filters
|
|
|
|
| 72 |
|
| 73 |
# ---------- Scoring-oriented thresholds ----------
|
| 74 |
# Low threshold for candidate generation
|
| 75 |
+
self.conf_thres = 0.25
|
| 76 |
|
| 77 |
# High-confidence boxes can survive without TTA confirmation
|
| 78 |
self.conf_high = 0.5
|
|
|
|
| 83 |
# TTA confirmation IoU
|
| 84 |
self.tta_match_iou = 0.55
|
| 85 |
|
| 86 |
+
self.max_det = 150
|
| 87 |
self.use_tta = True
|
| 88 |
|
| 89 |
# Box sanity filters
|