stfkramer
/

ScoreVision-v2

Model card Files Files and versions

stfkramer commited on 28 days ago

Commit

681e9ca

·

verified ·

1 Parent(s): 24e78ac

Update miner.py

Files changed (1) hide show

miner.py +1 -1

miner.py CHANGED Viewed

@@ -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", "bat", "glove", "graffiti", "hoodie", "spray paint"]
     input_size = 1280

     # (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"]
     input_size = 1280