coolroman commited on
Commit
02a8884
·
verified ·
1 Parent(s): 3013f3c

v2.0: hermestech weights + tile s<3 + include 3090/4090

Browse files
Files changed (4) hide show
  1. chute_config.yml +1 -1
  2. miner.py +11 -15
  3. weights.onnx +2 -2
  4. weights_alfred_backup.onnx +3 -0
chute_config.yml CHANGED
@@ -8,7 +8,7 @@ Image:
8
  NodeSelector:
9
  gpu_count: 1
10
  min_vram_gb_per_gpu: 16
11
- include:
12
  - "3090"
13
  - "4090"
14
 
 
8
  NodeSelector:
9
  gpu_count: 1
10
  min_vram_gb_per_gpu: 16
11
+ include_gpu_type:
12
  - "3090"
13
  - "4090"
14
 
miner.py CHANGED
@@ -1,16 +1,12 @@
1
- """Plate-detection miner — v1.4 "sparse tile, no flip — lat-trimmed".
2
 
3
- Base weights: alfred8995/arabic002 (YOLO26s @ 1280, fp16, ~19 MB).
4
 
5
  Inference pipeline:
6
- 1) Full-image primary pass with arabic002's strict tuning
7
- (conf=0.27, iou=0.444, sigma=0.5, soft-NMS + hflip TTA, max_det=18).
8
- 2) If the primary returned fewer than SPARSE_THRESHOLD (5) boxes,
9
- run a 2x2 overlapping tile pass with higher conf (tile_conf=0.40)
10
- and novelty-merge: keep a tile box only when it does not overlap
11
- any primary box at IoU >= 0.10. Tile augmentation is skipped
12
- entirely on challenges where the primary already has enough
13
- detections, so the FP score stays intact on dense scenes.
14
  """
15
  from pathlib import Path
16
  import math
@@ -95,11 +91,11 @@ class Miner:
95
  self.input_height = self._safe_dim(self.input_shape[2], default=SIZE)
96
  self.input_width = self._safe_dim(self.input_shape[3], default=SIZE)
97
 
98
- # Primary pass: arabic002 strict tuning
99
- self.conf_thres = 0.27
100
- self.iou_thres = 0.444
101
- self.sigma = 0.5
102
- self.max_det = 18
103
 
104
  # Conditional tile-pass (trimmed for latency: no hflip, tighter sparse)
105
  self.sparse_threshold = 3 # fire tiles only if primary returns < this
 
1
+ """Plate-detection miner — v2.0 "hermestech + tile s<3".
2
 
3
+ Base weights: hermestech00/numberplate0 (YOLO26s retrained, fp16, ~19 MB).
4
 
5
  Inference pipeline:
6
+ 1) Full-image primary pass with alfred001 tuning
7
+ (conf=0.22, iou=0.41, sigma=0.685, soft-NMS + hflip TTA).
8
+ 2) If the primary returned fewer than 3 boxes, run a 2x2
9
+ overlapping tile pass (tile_conf=0.40) with novelty-merge.
 
 
 
 
10
  """
11
  from pathlib import Path
12
  import math
 
91
  self.input_height = self._safe_dim(self.input_shape[2], default=SIZE)
92
  self.input_width = self._safe_dim(self.input_shape[3], default=SIZE)
93
 
94
+ # Primary pass: alfred001 tuning (optimized for hermestech weights)
95
+ self.conf_thres = 0.22
96
+ self.iou_thres = 0.41
97
+ self.sigma = 0.685
98
+ self.max_det = 300
99
 
100
  # Conditional tile-pass (trimmed for latency: no hflip, tighter sparse)
101
  self.sparse_threshold = 3 # fire tiles only if primary returns < this
weights.onnx CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:85a9c463abc53cffa7e5607ab1a4ba5e7d60106fee911644e6aec238d436963e
3
- size 19388678
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ac86065b9d2b73ee302124f3fdaae9144d09d756ab8290900146e98375bba724
3
+ size 19892467
weights_alfred_backup.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:85a9c463abc53cffa7e5607ab1a4ba5e7d60106fee911644e6aec238d436963e
3
+ size 19388678