scorevision: push artifact
Browse files- miner.py +8 -7
- person_weights.onnx +2 -2
miner.py
CHANGED
|
@@ -1,19 +1,20 @@
|
|
| 1 |
"""
|
| 2 |
-
Score Vision SN44 β Unified miner v3 (2026-
|
| 3 |
-
Dual-model: vehicle (YOLO11m INT8 1280) + person (
|
| 4 |
Vehicle weights loaded from secondary HF repo (meaculpitt/ScoreVision-Vehicle).
|
| 5 |
Person weights loaded from primary HF repo (template downloads automatically).
|
| 6 |
|
| 7 |
Vehicle model (vehicle_weights.onnx):
|
| 8 |
Trained classes: 0=car, 1=bus, 2=truck, 3=motorcycle
|
| 9 |
-
Remapped to
|
|
|
|
| 10 |
|
| 11 |
Person model (person_weights.onnx):
|
| 12 |
-
Single class: 0=person
|
| 13 |
|
| 14 |
Both models run on every image. All detections merged.
|
| 15 |
-
cls_id
|
| 16 |
-
Vehicle eval uses cls_id
|
| 17 |
"""
|
| 18 |
|
| 19 |
import os
|
|
@@ -161,7 +162,7 @@ from concurrent.futures import ThreadPoolExecutor, as_completed
|
|
| 161 |
logger = logging.getLogger(__name__)
|
| 162 |
|
| 163 |
# ββ Vehicle config ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 164 |
-
VEH_MODEL_TO_OUT: dict[int, int] = {0: 1, 1:
|
| 165 |
VEH_NUM_CLASSES = 4
|
| 166 |
VEH_CONF_THRES = 0.35
|
| 167 |
VEH_TTA_CONF = 0.10
|
|
|
|
| 1 |
"""
|
| 2 |
+
Score Vision SN44 β Unified miner v3.5 (2026-04-02).
|
| 3 |
+
Dual-model: vehicle (YOLO11m INT8 1280) + person (YOLO26s FP16 960 end2end).
|
| 4 |
Vehicle weights loaded from secondary HF repo (meaculpitt/ScoreVision-Vehicle).
|
| 5 |
Person weights loaded from primary HF repo (template downloads automatically).
|
| 6 |
|
| 7 |
Vehicle model (vehicle_weights.onnx):
|
| 8 |
Trained classes: 0=car, 1=bus, 2=truck, 3=motorcycle
|
| 9 |
+
Remapped to output: 1=car, 4=bus(filtered), 2=truck, 3=motorcycle
|
| 10 |
+
Bus remapped to cls_id=4 to avoid collision with person cls_id=0.
|
| 11 |
|
| 12 |
Person model (person_weights.onnx):
|
| 13 |
+
YOLO26s FP16 960px end2end [1,300,6]. Single class: 0=person.
|
| 14 |
|
| 15 |
Both models run on every image. All detections merged.
|
| 16 |
+
Vehicle cls_id=4 (bus) filtered by validator (out of range for both elements).
|
| 17 |
+
Vehicle eval uses cls_id 1-3. Person eval uses cls_id 0 only.
|
| 18 |
"""
|
| 19 |
|
| 20 |
import os
|
|
|
|
| 162 |
logger = logging.getLogger(__name__)
|
| 163 |
|
| 164 |
# ββ Vehicle config ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 165 |
+
VEH_MODEL_TO_OUT: dict[int, int] = {0: 1, 1: 4, 2: 2, 3: 3} # busβ4 (avoid person cls_id=0 collision)
|
| 166 |
VEH_NUM_CLASSES = 4
|
| 167 |
VEH_CONF_THRES = 0.35
|
| 168 |
VEH_TTA_CONF = 0.10
|
person_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:b23ca9c0a7efd6613753b66033d7c6637b338a79547562687d5e58ba5fb92d7c
|
| 3 |
+
size 19258183
|