Upload folder using huggingface_hub

football_object_detection.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:86033baaab62ecf30c31934e5d21f3115af6d33c23b1bb0e454f917302cf9956
+size 38568341

miner.py

@@ -49,12 +49,12 @@ class Miner:
     CORNER_CONFIDENCE = Constants.CORNER_CONFIDENCE
     GOALKEEPER_POSITION_MARGIN = Constants.GOALKEEPER_POSITION_MARGIN
     MIN_SAMPLES_FOR_FIT = 16  # Minimum player crops needed before fitting TeamClassifier
-    MAX_SAMPLES_FOR_FIT = 600  # Maximum samples to avoid overfitting
+    MAX_SAMPLES_FOR_FIT = 1000  # Maximum samples to avoid overfitting
 
     def __init__(self, path_hf_repo: Path) -> None:
         try:
             device = "cuda" if torch.cuda.is_available() else "cpu"
-            model_path = path_hf_repo / "detection.onnx"
+            model_path = path_hf_repo / "football_object_detection.onnx"
             self.bbox_model = YOLO(model_path)
             
             print(f"BBox Model Loaded: class name {self.bbox_model.names}")