Update app.py

app.py

@@ -22,6 +22,9 @@ PITCH_WIDTH = 3.05    # Width of pitch
 STUMP_HEIGHT = 0.71   # Stump height
 STUMP_WIDTH = 0.2286  # Stump width (including bails)
 
+# Model input size (adjust if best.pt was trained with a different size)
+MODEL_INPUT_SIZE = (640, 640)  # (height, width)
+
 # Load model
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = attempt_load("best.pt")  # Load without map_location
@@ -44,6 +47,9 @@ def process_video(video_path):
         if not ret:
             break
 
+        # Resize frame to model input size
+        frame = cv2.resize(frame, MODEL_INPUT_SIZE, interpolation=cv2.INTER_AREA)
+
         # Preprocess frame for YOLOv5
         img = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
         img = torch.from_numpy(img).to(device).float() / 255.0
@@ -61,6 +67,9 @@ def process_video(video_path):
                 for *xyxy, conf, cls in det:
                     x_center = (xyxy[0] + xyxy[2]) / 2
                     y_center = (xyxy[1] + xyxy[3]) / 2
+                    # Scale coordinates back to original frame size
+                    x_center = x_center * frame_width / MODEL_INPUT_SIZE[1]
+                    y_center = y_center * frame_height / MODEL_INPUT_SIZE[0]
                     positions.append((x_center.item(), y_center.item()))
                     frame_numbers.append(frame_num)