Upload folder using huggingface_hub
Browse files- chute_config.yml +8 -0
- miner.py +3 -3
- weights.onnx +2 -2
chute_config.yml
CHANGED
|
@@ -11,6 +11,14 @@ NodeSelector:
|
|
| 11 |
min_memory_gb: 32
|
| 12 |
min_cpu_count: 32
|
| 13 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 14 |
Chute:
|
| 15 |
timeout_seconds: 900
|
| 16 |
concurrency: 4
|
|
|
|
| 11 |
min_memory_gb: 32
|
| 12 |
min_cpu_count: 32
|
| 13 |
|
| 14 |
+
exclude:
|
| 15 |
+
- "5090"
|
| 16 |
+
- A100
|
| 17 |
+
- h100
|
| 18 |
+
- b200
|
| 19 |
+
- h200
|
| 20 |
+
- mi300x
|
| 21 |
+
|
| 22 |
Chute:
|
| 23 |
timeout_seconds: 900
|
| 24 |
concurrency: 4
|
miner.py
CHANGED
|
@@ -73,15 +73,15 @@ class Miner:
|
|
| 73 |
|
| 74 |
# Tuned for validator scoring: reduce FP (FALSE_POSITIVE pillar),
|
| 75 |
# preserve recall (MAP50, RECALL), improve precision.
|
| 76 |
-
self.conf_thres = 0.
|
| 77 |
self.iou_thres = 0.5 # Lower = suppress duplicate detections (FP)
|
| 78 |
self.max_det = 150 # Cap detections; sports ~20-30 persons
|
| 79 |
self.use_tta = True
|
| 80 |
|
| 81 |
# Box sanity: filter tiny/spurious detections (common FP source)
|
| 82 |
-
self.min_box_area =
|
| 83 |
self.min_side = 8
|
| 84 |
-
self.max_aspect_ratio =
|
| 85 |
|
| 86 |
print(f"✅ ONNX model loaded from: {model_path}")
|
| 87 |
print(f"✅ ONNX providers: {self.session.get_providers()}")
|
|
|
|
| 73 |
|
| 74 |
# Tuned for validator scoring: reduce FP (FALSE_POSITIVE pillar),
|
| 75 |
# preserve recall (MAP50, RECALL), improve precision.
|
| 76 |
+
self.conf_thres = 0.33 # Higher = fewer FP, slightly lower recall
|
| 77 |
self.iou_thres = 0.5 # Lower = suppress duplicate detections (FP)
|
| 78 |
self.max_det = 150 # Cap detections; sports ~20-30 persons
|
| 79 |
self.use_tta = True
|
| 80 |
|
| 81 |
# Box sanity: filter tiny/spurious detections (common FP source)
|
| 82 |
+
self.min_box_area = 12 * 12 # ~144 px²
|
| 83 |
self.min_side = 8
|
| 84 |
+
self.max_aspect_ratio = 10.0
|
| 85 |
|
| 86 |
print(f"✅ ONNX model loaded from: {model_path}")
|
| 87 |
print(f"✅ ONNX providers: {self.session.get_providers()}")
|
weights.onnx
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7e822fc5cebb28333419e17b7eb2d26fbb0f006347d9e9b9261d78b7a20aed2c
|
| 3 |
+
size 19720415
|