SuperBitDev commited on
Commit
7289e14
·
verified ·
1 Parent(s): c02ef35

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. chute_config.yml +8 -0
  2. miner.py +3 -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.35 # 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 = 14 * 14 # ~144 px²
83
  self.min_side = 8
84
- self.max_aspect_ratio = 8.0
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:4a7fac9f674924560d7a4788e2e1a4733b1055e4380f9e72e0720ed9fb3d155b
3
- size 19404973
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7e822fc5cebb28333419e17b7eb2d26fbb0f006347d9e9b9261d78b7a20aed2c
3
+ size 19720415