meaculpitt commited on
Commit
822d8ce
Β·
verified Β·
1 Parent(s): 2005c61

scorevision: push artifact

Browse files
Files changed (1) hide show
  1. miner.py +5 -5
miner.py CHANGED
@@ -2,8 +2,8 @@
2
  Score Vision SN44 β€” VehicleDetect miner endpoint.
3
 
4
  Class mapping (output indices):
5
- 0 = car (COCO class 2)
6
- 1 = bus (COCO class 5)
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] = {2: 0, 5: 1, 7: 2, 3: 3}
62
  COCO_VEHICLE_IDX = list(COCO_TO_OUT.keys())
63
- OUT_NAMES = ["car", "bus", "truck", "motorcycle"]
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