Upload 6 files

player_ball_assigner/__init__.py

@@ -0,0 +1 @@
+from .player_ball_assigner import PlayerBallAssigner

player_ball_assigner/player_ball_assigner.py

@@ -0,0 +1,27 @@
+import sys 
+sys.path.append('../')
+from utils import get_center_of_bbox, measure_distance
+
+class PlayerBallAssigner():
+    def __init__(self):
+        self.max_player_ball_distance = 70
+    
+    def assign_ball_to_player(self,players,ball_bbox):
+        ball_position = get_center_of_bbox(ball_bbox)
+
+        miniumum_distance = 99999
+        assigned_player=-1
+
+        for player_id, player in players.items():
+            player_bbox = player['bbox']
+
+            distance_left = measure_distance((player_bbox[0],player_bbox[-1]),ball_position)
+            distance_right = measure_distance((player_bbox[2],player_bbox[-1]),ball_position)
+            distance = min(distance_left,distance_right)
+
+            if distance < self.max_player_ball_distance:
+                if distance < miniumum_distance:
+                    miniumum_distance = distance
+                    assigned_player = player_id
+
+        return assigned_player