Upload folder using huggingface_hub

miner.py

@@ -34,7 +34,7 @@ class Miner:
     # (ships inside weights.onnx), so a retrained model with a different class
     # order is remapped correctly without code changes. This static list is used
     # only when that metadata is missing or unparsable.
-    model_class_names = ["balaclava", "hoodie", "glove", "bat", "spray paint", "graffiti"]
+    model_class_names = ["balaclava", "bat", "glove", "graffiti", "hoodie", "spray paint"]
 
     input_size = 1280
 
@@ -50,13 +50,14 @@ class Miner:
     max_det = 150
     
     _conf_thres_array = np.array(
-        [0.25, 0.26, 0.22, 0.21, 0.3, 0.28], dtype=np.float32,
+        [0.38, 0.42, 0.22, 0.12, 0.33, 0.20], dtype=np.float32,
     )
     _bonus_array = np.array(
-        [0.05, 0.04, 0.04, 0.01, 0.08, 0.08], dtype=np.float32,
+        [0.2, 0.25, 0.12, 0.09, 0.21, 0.06], dtype=np.float32,
     )
-
+    
     def __init__(self, path_hf_repo: Path) -> None:
+        # model_path = path_hf_repo / "weights-all.onnx"
         model_path = path_hf_repo / "weights.onnx"
         print("ORT version:", ort.__version__)
 
@@ -141,7 +142,6 @@ class Miner:
 
     def _read_model_class_order(self) -> "list[str] | None":
         """Read the model's class order from Ultralytics ONNX metadata.
-
         Returns the class names ordered by model-emit index, or None when the
         metadata is missing/unparsable or doesn't match `class_names` as a set
         (in which case the static model_class_names fallback is used)."""

weights.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f9ff74df8edf5ed276720ed2ac42dafd8e9994d1bacb94d215bab1068b8e3cbc
-size 9809125
+oid sha256:a9705e464e581151416bd2dffc5b4dc7ff1cc4f3b899fee82aeb692a004a9e2e
+size 9809235