v2.0: hermestech weights + tile s<3 + include 3090/4090
Browse files- chute_config.yml +1 -1
- miner.py +11 -15
- weights.onnx +2 -2
- 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 |
-
|
| 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 —
|
| 2 |
|
| 3 |
-
Base weights:
|
| 4 |
|
| 5 |
Inference pipeline:
|
| 6 |
-
1) Full-image primary pass with
|
| 7 |
-
(conf=0.
|
| 8 |
-
2) If the primary returned fewer than
|
| 9 |
-
|
| 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:
|
| 99 |
-
self.conf_thres = 0.
|
| 100 |
-
self.iou_thres = 0.
|
| 101 |
-
self.sigma = 0.
|
| 102 |
-
self.max_det =
|
| 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:
|
| 3 |
-
size
|
|
|
|
| 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
|