initial commit

inference/hsv_classifier.py

@@ -358,7 +358,7 @@ def add_non_black_pixels_count_in_filter(
 
     def predict_img(self, img: np.ndarray) -> str:
         """
-        Gets the filter with most non blakc pixels on img and returns its name.
+        Gets the filter with most non black pixels on img and returns its name.
 
         Parameters
         ----------

soccer/match.py

@@ -37,7 +37,7 @@ def __init__(self, home: Team, away: Team, fps: int = 30):
         # Amount of consecutive frames new team has to have the ball in order to change possession
         self.possesion_counter_threshold = 20
         # Distance in pixels from player to ball in order to consider a player has the ball
-        self.ball_distance_threshold = 45
+        self.ball_distance_threshold = 30
         self.fps = fps
         # Pass detection
         self.pass_event = PassEvent()

soccer/pass_event.py

@@ -212,7 +212,7 @@ def __init__(self) -> None:
         self.init_player_with_ball = None
         self.last_player_with_ball = None
         self.player_with_ball_counter = 0
-        self.player_with_ball_threshold = 3
+        self.player_with_ball_threshold = 1
         self.player_with_ball_threshold_dif_team = 4
 
     def update(self, closest_player: Player, ball: Ball) -> None: