Update app.py

app.py

@@ -4,7 +4,7 @@ import numpy as np
 import torch
 from ultralytics import YOLO
 import gradio as gr
-from scipy.interpolate import interp1d
+from scipy.interpolate import interp1d, UnivariateSpline
 import uuid
 import os
 
@@ -16,9 +16,9 @@ STUMPS_WIDTH = 0.2286  # meters (width of stumps)
 FRAME_RATE = 20  # Input video frame rate
 SLOW_MOTION_FACTOR = 2  # Reduced for faster output
 CONF_THRESHOLD = 0.25  # Confidence threshold for detection
-PITCH_ZONE_Y = 0.9  # Fraction of frame height for pitch zone
-IMPACT_ZONE_Y = 0.8  # Fraction of frame height for impact zone
-IMPACT_DELTA_Y = 50  # Pixels for detecting sudden y-position change
+PITCH_ZONE_Y = 0.85  # Adjusted for pitch near stumps
+IMPACT_ZONE_Y = 0.75  # Adjusted for impact near batsman leg
+IMPACT_DELTA_Y = 30  # Reduced for finer impact detection
 STUMPS_HEIGHT = 0.711  # meters (height of stumps)
 
 def process_video(video_path):
@@ -61,7 +61,7 @@ def estimate_trajectory(ball_positions, detection_frames, frames):
         return None, None, None, None, None, None, "Error: Fewer than 2 valid single-ball detections for trajectory"
     frame_height = frames[0].shape[0]
 
-    # Filter to unique positions to reduce interpolation points
+    # Filter to unique positions
     unique_positions = [ball_positions[0]]
     for pos in ball_positions[1:]:
         if abs(pos[0] - unique_positions[-1][0]) > 10 or abs(pos[1] - unique_positions[-1][1]) > 10:
@@ -70,6 +70,10 @@ def estimate_trajectory(ball_positions, detection_frames, frames):
     y_coords = [pos[1] for pos in unique_positions]
     times = np.array([i / FRAME_RATE for i in range(len(unique_positions))])
 
+    # Smooth coordinates with spline interpolation
+    x_smooth = UnivariateSpline(times, x_coords, s=10)
+    y_smooth = UnivariateSpline(times, y_coords, s=10)
+
     pitch_idx = 0
     for i, y in enumerate(y_coords):
         if y > frame_height * PITCH_ZONE_Y:
@@ -94,12 +98,12 @@ def estimate_trajectory(ball_positions, detection_frames, frames):
     times = times[:impact_idx + 1]
 
     try:
-        fx = interp1d(times, x_coords, kind='linear', fill_value="extrapolate")
-        fy = interp1d(times, y_coords, kind='quadratic', fill_value="extrapolate")
+        fx = interp1d(times, x_smooth(times), kind='linear', fill_value="extrapolate")
+        fy = interp1d(times, y_smooth(times), kind='quadratic', fill_value="extrapolate")
     except Exception as e:
         return None, None, None, None, None, None, f"Error in trajectory interpolation: {str(e)}"
 
-    vis_trajectory = list(zip(x_coords, y_coords))
+    vis_trajectory = list(zip(x_smooth(times), y_smooth(times)))
     t_full = np.linspace(times[0], times[-1] + 0.5, len(times) + 5)
     x_full = fx(t_full)
     y_full = fy(t_full)
@@ -118,9 +122,9 @@ def lbw_decision(ball_positions, full_trajectory, frames, pitch_point, impact_po
 
     frame_height, frame_width = frames[0].shape[:2]
     stumps_x = frame_width / 2
-    stumps_y = frame_height * 0.9
+    stumps_y = frame_height * 0.85  # Adjusted to align with pitch
     stumps_width_pixels = frame_width * (STUMPS_WIDTH / 3.0)
-    batsman_area_y = frame_height * 0.8
+    batsman_area_y = frame_height * 0.75
 
     pitch_x, pitch_y = pitch_point
     impact_x, impact_y = impact_point
@@ -150,7 +154,7 @@ def generate_slow_motion(frames, vis_trajectory, pitch_point, pitch_frame, impac
         return None
     frame_height, frame_width = frames[0].shape[:2]
     stumps_x = frame_width / 2
-    stumps_y = frame_height * 0.9
+    stumps_y = frame_height * 0.85  # Align with pitch
     stumps_width_pixels = frame_width * (STUMPS_WIDTH / 3.0)
     stumps_height_pixels = frame_height * (STUMPS_HEIGHT / 3.0)
 
@@ -160,33 +164,38 @@ def generate_slow_motion(frames, vis_trajectory, pitch_point, pitch_frame, impac
     trajectory_points = np.array(vis_trajectory, dtype=np.int32).reshape((-1, 1, 2))
 
     for i, frame in enumerate(frames):
-        # Draw stumps (single line for efficiency)
+        # Draw stumps outline
         cv2.line(frame, (int(stumps_x - stumps_width_pixels / 2), int(stumps_y)),
                  (int(stumps_x + stumps_width_pixels / 2), int(stumps_y)), (255, 255, 255), 2)
+        cv2.line(frame, (int(stumps_x - stumps_width_pixels / 2), int(stumps_y - stumps_height_pixels)),
+                 (int(stumps_x - stumps_width_pixels / 2), int(stumps_y)), (255, 255, 255), 2)
+        cv2.line(frame, (int(stumps_x + stumps_width_pixels / 2), int(stumps_y - stumps_height_pixels)),
+                 (int(stumps_x + stumps_width_pixels / 2), int(stumps_y)), (255, 255, 255), 2)
 
-        # Draw crease line
-        cv2.line(frame, (0, int(stumps_y)), (frame_width, int(stumps_y)), (255, 255, 0), 2)
+        # Draw crease line at stumps
+        cv2.line(frame, (int(stumps_x - stumps_width_pixels / 2), int(stumps_y)),
+                 (int(stumps_x + stumps_width_pixels / 2), int(stumps_y)), (255, 255, 0), 2)
 
         if i in detection_frames and trajectory_points.size > 0:
             idx = detection_frames.index(i) + 1
             if idx <= len(trajectory_points):
-                cv2.polylines(frame, [trajectory_points[:idx]], False, (255, 0, 0), 2)
+                cv2.polylines(frame, [trajectory_points[:idx]], False, (0, 0, 255), 2)  # Blue trajectory
 
         if pitch_point and i == pitch_frame:
             x, y = pitch_point
-            cv2.circle(frame, (int(x), int(y)), 8, (0, 255, 0), -1)
-            cv2.putText(frame, "Pitching Factor", (int(x) + 10, int(y) - 10),
+            cv2.circle(frame, (int(x), int(y)), 8, (0, 255, 0), -1)  # Green for pitching
+            cv2.putText(frame, "Pitching", (int(x) + 10, int(y) - 10),
                         cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 255, 0), 1)
 
         if impact_point and i == impact_frame:
             x, y = impact_point
-            cv2.circle(frame, (int(x), int(y)), 8, (0, 0, 255), -1)
-            cv2.putText(frame, "Impact Factor", (int(x) + 10, int(y) + 20),
+            cv2.circle(frame, (int(x), int(y)), 8, (0, 0, 255), -1)  # Red for impact
+            cv2.putText(frame, "Impact", (int(x) + 10, int(y) + 20),
                         cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 0, 255), 1)
 
         if impact_point and i == impact_frame and "Out" in decision:
-            cv2.putText(frame, "Wicket Factor", (int(stumps_x) - 50, int(stumps_y) - 20),
-                        cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 165, 255), 1)
+            cv2.putText(frame, "Wickets", (int(stumps_x) - 50, int(stumps_y) - 20),
+                        cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 165, 255), 1)  # Orange for wickets
 
         for _ in range(SLOW_MOTION_FACTOR):
             out.write(frame)
@@ -212,10 +221,10 @@ iface = gr.Interface(
     inputs=gr.Video(label="Upload Video Clip"),
     outputs=[
         gr.Textbox(label="DRS Decision and Debug Log"),
-        gr.Video(label="Optimized Slow-Motion Replay with Pitching Factor (Green), Impact Factor (Red), Wicket Factor (Orange), Stumps (White), Crease (Yellow)")
+        gr.Video(label="Optimized Slow-Motion Replay with Pitching (Green), Impact (Red), Wickets (Orange), Stumps (White), Crease (Yellow)")
     ],
     title="AI-Powered DRS for LBW in Local Cricket",
-    description="Upload a video clip of a cricket delivery to get an LBW decision and optimized slow-motion replay showing pitching factor (green circle), impact factor (red circle), wicket factor (orange text), stumps (white lines), and crease line (yellow line)."
+    description="Upload a video clip of a cricket delivery to get an LBW decision and optimized slow-motion replay showing pitching (green circle), impact (red circle), wickets (orange text), stumps (white outline), and crease line (yellow line)."
 )
 
 if __name__ == "__main__":