video

app.py

@@ -11,13 +11,17 @@ from view_transformer import ViewTransformer
 from speed_and_distance_estimator import SpeedAndDistance_Estimator
 
 def process_video(input_video):
+    print(input_video)
     # Read Video
     video_frames = read_video(input_video.name)  # use the uploaded file
-
+    
     # Initialize Tracker
     tracker = Tracker('models/best.pt')
 
-    tracks = tracker.get_object_tracks(video_frames)
+    #tracks = tracker.get_object_tracks(video_frames)
+    tracks = tracker.get_object_tracks(video_frames,
+                                       read_from_stub=True,
+                                       stub_path='stubs/track_stub_121364_0_small.pkl')
 
     # Interpolate Ball Positions
     tracks["ball"] = tracker.interpolate_ball_positions(tracks["ball"])
@@ -27,7 +31,10 @@ def process_video(input_video):
 
     # Camera movement estimator
     camera_movement_estimator = CameraMovementEstimator(video_frames[0])
-    camera_movement_per_frame = camera_movement_estimator.get_camera_movement(video_frames)
+    #camera_movement_per_frame = camera_movement_estimator.get_camera_movement(video_frames)
+    camera_movement_per_frame = camera_movement_estimator.get_camera_movement(video_frames,
+                                                                                read_from_stub=True,
+                                                                                stub_path='stubs/camera_movement_stub_121364_0_small.pkl')
     camera_movement_estimator.add_adjust_positions_to_tracks(tracks, camera_movement_per_frame)
 
     # View Transformer
@@ -77,8 +84,8 @@ def process_video(input_video):
 def create_gradio_interface():
     # Input: video file, Output: video file
     interface = gr.Interface(fn=process_video,
-                             inputs=gr.Video(type="file", label="Upload Video (mp4, avi, mov)"),
-                             outputs=gr.Video(type="file", label="Processed Video"))
+                             inputs=gr.Video(label="Upload Video (mp4, avi, mov)"),
+                             outputs=gr.Video(label="Processed Video"))
     
     return interface