Update app.py

app.py

@@ -24,7 +24,7 @@ PITCH_LENGTH = 20.12  # meters (standard cricket pitch length)
 STUMPS_HEIGHT = 0.71  # meters (stumps height)
 CAMERA_HEIGHT = 2.0  # meters (assumed camera height)
 CAMERA_DISTANCE = 10.0  # meters (assumed camera distance from pitch)
-MAX_POSITION_JUMP = 100  # Pixels, threshold for unexpected position changes
+MAX_POSITION_JUMP = 50  # Pixels, threshold for unexpected position changes
 
 def process_video(video_path):
     if not os.path.exists(video_path):
@@ -82,16 +82,19 @@ def estimate_trajectory(ball_positions, frames, detection_frames):
         return None, None, None, None, None, None, None, None, None, "Error: Fewer than 2 ball detections for trajectory"
     frame_height, frame_width = frames[0].shape[:2]
 
-    # Filter out unexpected jumps in position
+    # Filter out sudden changes in position for continuous trajectory
     filtered_positions = [ball_positions[0]]
     filtered_frames = [detection_frames[0]]
     for i in range(1, len(ball_positions)):
-        prev_pos = ball_positions[i-1]
+        prev_pos = filtered_positions[-1]
         curr_pos = ball_positions[i]
         distance = np.sqrt((curr_pos[0] - prev_pos[0])**2 + (curr_pos[1] - prev_pos[1])**2)
         if distance <= MAX_POSITION_JUMP:
             filtered_positions.append(curr_pos)
             filtered_frames.append(detection_frames[i])
+        else:
+            # Skip sudden jumps to maintain continuity
+            continue
 
     if len(filtered_positions) < 2:
         return None, None, None, None, None, None, None, None, None, "Error: Fewer than 2 valid ball detections after filtering"
@@ -122,12 +125,13 @@ def estimate_trajectory(ball_positions, frames, detection_frames):
     impact_point = filtered_positions[impact_idx]
 
     try:
-        fx = interp1d(times[:impact_idx + 1], x_coords[:impact_idx + 1], kind='linear', fill_value="extrapolate")
-        fy = interp1d(times[:impact_idx + 1], y_coords[:impact_idx + 1], kind='quadratic', fill_value="extrapolate")
+        # Use cubic interpolation for smoother trajectory
+        fx = interp1d(times[:impact_idx + 1], x_coords[:impact_idx + 1], kind='cubic', fill_value="extrapolate")
+        fy = interp1d(times[:impact_idx + 1], y_coords[:impact_idx + 1], kind='cubic', fill_value="extrapolate")
     except Exception as e:
         return None, None, None, None, None, None, None, None, None, f"Error in trajectory interpolation: {str(e)}"
 
-    t_full = np.linspace(times[0], times[-1] + 0.5, len(times) * 2)  # Smoother trajectory
+    t_full = np.linspace(times[0], times[-1], len(times) * 4)  # Dense points for smooth trajectory
     x_full = fx(t_full)
     y_full = fy(t_full)
     trajectory_2d = list(zip(x_full, y_full))