Upload folder using huggingface_hub

miner.py

@@ -113,7 +113,7 @@ class Miner:
         self.max_det = 200
         # TTA = a 2nd (flipped) forward pass. Doubles latency; off for the
         # CPU latency gate. Re-enable only if the latency budget allows.
-        self.use_tta = True
+        self.use_tta = False
 
 #       conf thresholds:  broom=0.38  drainage gate=0.45  nozzle=0.30  track=0.60
         # Per-class confidence thresholds.
@@ -123,7 +123,7 @@ class Miner:
         # 0.5/0.5 silently discarded many valid detections); track is the
         # one class where false fires are common enough to need 0.38.
         self._conf_thres_array = np.array(
-            [0.28, 0.38, 0.60, 0.45], dtype=np.float32
+            [0.32, 0.3, 0.45, 0.3], dtype=np.float32
         )
         # Per-class rescue bonus: when a class has ZERO boxes passing the
         # threshold in a frame, its top-1 candidate is admitted when its score
@@ -131,7 +131,7 @@ class Miner:
         # DISABLED (all zeros): the sweep showed rescue admits more false
         # positives than true positives under the validator's FP pillar.
         self._bonus_array = np.array(
-            [0.05, 0.1, 0.25, 0.2], dtype=np.float32
+            [0.05, 0.1, 0.1, 0.1], dtype=np.float32
         )
 
         # Box sanity filter — kept loose: car-wash `nozzle` boxes are tiny
@@ -145,6 +145,17 @@ class Miner:
         print(f"✅ ONNX providers: {self.session.get_providers()}")
         print(f"✅ ONNX input: name={self.input_name}, shape={self.input_shape}")
 
+        self._warmup()
+
+    def _warmup(self, iters: int = 3) -> None:
+        try:
+            dummy = np.zeros((720, 1280, 3), dtype=np.uint8)
+            for _ in range(max(1, iters)):
+                self.predict_batch(batch_images=[dummy], offset=0, n_keypoints=0)
+            print(f"✅ warmup: {iters} dummy predict_batch call(s) done")
+        except Exception as e:
+            print(f"⚠️ warmup skipped: {e}")
+
     def __repr__(self) -> str:
         return (
             f"ONNXRuntime(session={type(self.session).__name__}, "

weights.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7d0089998a8868844db5012f76470768e2cc92eeee92a2c22e84c46fd4109ea5
-size 19287011
+oid sha256:020517c8e4ddd15a3a655fc794aa69bbda6032b328feb2749ef917c92b27b50f
+size 9881386