scorevision: push artifact
Browse files
miner.py
CHANGED
|
@@ -2,8 +2,8 @@
|
|
| 2 |
Score Vision SN44 β VehicleDetect miner endpoint.
|
| 3 |
|
| 4 |
Class mapping (output indices):
|
| 5 |
-
0 =
|
| 6 |
-
1 =
|
| 7 |
2 = truck (COCO class 7)
|
| 8 |
3 = motorcycle (COCO class 3)
|
| 9 |
|
|
@@ -57,10 +57,10 @@ IMG_SIZE = 640
|
|
| 57 |
CONF_THRESH = 0.55 # sweep: max composite score (0.60ΓmAP + 0.40ΓFP_score) at conf=0.55
|
| 58 |
IOU_THRESH = 0.45
|
| 59 |
|
| 60 |
-
# COCO class index β submission class index
|
| 61 |
-
COCO_TO_OUT: dict[int, int] = {
|
| 62 |
COCO_VEHICLE_IDX = list(COCO_TO_OUT.keys())
|
| 63 |
-
OUT_NAMES = ["
|
| 64 |
|
| 65 |
# ββ Model loader (singleton) βββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 66 |
_SESSION: ort.InferenceSession | None = None
|
|
|
|
| 2 |
Score Vision SN44 β VehicleDetect miner endpoint.
|
| 3 |
|
| 4 |
Class mapping (output indices):
|
| 5 |
+
0 = bus (COCO class 5)
|
| 6 |
+
1 = car (COCO class 2)
|
| 7 |
2 = truck (COCO class 7)
|
| 8 |
3 = motorcycle (COCO class 3)
|
| 9 |
|
|
|
|
| 57 |
CONF_THRESH = 0.55 # sweep: max composite score (0.60ΓmAP + 0.40ΓFP_score) at conf=0.55
|
| 58 |
IOU_THRESH = 0.45
|
| 59 |
|
| 60 |
+
# COCO class index β submission class index (official order: 0=bus, 1=car, 2=truck, 3=motorcycle)
|
| 61 |
+
COCO_TO_OUT: dict[int, int] = {5: 0, 2: 1, 7: 2, 3: 3}
|
| 62 |
COCO_VEHICLE_IDX = list(COCO_TO_OUT.keys())
|
| 63 |
+
OUT_NAMES = ["bus", "car", "truck", "motorcycle"]
|
| 64 |
|
| 65 |
# ββ Model loader (singleton) βββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 66 |
_SESSION: ort.InferenceSession | None = None
|