backend

app.py

@@ -386,36 +386,22 @@ def update_tracker_model(model_name):
 def get_video_settings(video_name):
     """Get video-specific settings based on video name"""
     video_settings = {
-        "blackswan": (50, 756, 3),
-        "bike-packing": (50, 756, 3),
-        "bmx-trees": (50, 756, 3),
-        "breakdance": (50, 756, 3),
-        "camel": (50, 756, 3),
-        "car-roundabout": (50, 756, 3),
-        "car-shadow": (50, 756, 3),
-        "cows": (50, 756, 3),
-        "dance-twirl": (50, 756, 3),
-        "dog": (50, 756, 3),
-        "dogs-jump": (50, 756, 3),
-        "drift-chicane": (50, 756, 3),
-        "drift-straight": (50, 756, 3),
-        "goat": (50, 756, 3),
-        "gold-fish": (50, 756, 3),
-        "horsejump-high": (50, 756, 3),
-        "india": (50, 756, 3),
-        "judo": (50, 756, 3),
-        "kite-surf": (50, 756, 3),
-        "lab-coat": (50, 756, 3),
-        "libby": (50, 756, 3),
-        "loading": (50, 756, 3),
-        "mbike-trick": (50, 756, 3),
-        "motocross-jump": (50, 756, 3),
-        "paragliding-launch": (50, 756, 3),
-        "parkour": (50, 756, 3),
-        "pigs": (50, 756, 3),
-        "scooter-black": (50, 756, 3),
-        "shooting": (50, 756, 3),
-        "soapbox": (50, 756, 3)
+        "kiss": (45, 700, 10),
+        "backpack": (40, 600, 2),
+        "kitchen": (60, 800, 3),
+        "pillow": (35, 500, 2),
+        "hockey": (45, 700, 2),
+        "drifting": (35, 1000, 6),
+        "ball": (45, 256, 6),
+        "ken_block_0": (45, 700, 2),
+        "ego_kc1": (45, 500, 4),
+        "vertical_place": (45, 500, 3),
+        "ego_teaser": (45, 1200, 10),
+        "robot_unitree": (45, 500, 4),
+        "droid_robot": (35, 400, 5),
+        "robot_2": (45, 256, 5),
+        "cinema_0": (45, 356, 5),
+        "cinema_1": (45, 756, 3),
     }
     
     return video_settings.get(video_name, (50, 756, 3))
@@ -508,7 +494,7 @@ with gr.Blocks(
     Welcome to SpatialTracker V2! This interface allows you to track objects in videos using advanced computer vision techniques.
     
     **Instructions:**
-    1. Upload a video file
+    1. Upload a video file or select from examples below
     2. Click on the object you want to track in the first frame
     3. Adjust tracking parameters if needed
     4. Click "Launch Visualization" to start tracking
@@ -519,16 +505,42 @@ with gr.Blocks(
     status_text = "🟢 Backend Connected" if BACKEND_AVAILABLE else "🟡 Running in Standalone Mode"
     gr.Markdown(f"**Status:** {status_text}")
     
+    # Example videos section - moved to top
+    with gr.Group():
+        gr.Markdown("### 📂 Example Videos")
+        gr.Markdown("Try these example videos to get started quickly:")
+        
+        # Define video_input here so it can be referenced in examples
+        video_input = gr.Video(
+            label="Upload Video or Select Example",
+            format="mp4"
+        )
+        
+        gr.Examples(
+            examples=[
+                ["examples/kiss.mp4"],
+                ["examples/backpack.mp4"],
+                ["examples/kitchen.mp4"],
+                ["examples/pillow.mp4"],
+                ["examples/hockey.mp4"],
+                ["examples/drifting.mp4"],
+                ["examples/ball.mp4"],
+                ["examples/ken_block_0.mp4"],
+                ["examples/ego_kc1.mp4"],
+                ["examples/vertical_place.mp4"],
+                ["examples/ego_teaser.mp4"],
+                ["examples/robot_unitree.mp4"],
+                ["examples/droid_robot.mp4"],
+                ["examples/robot_2.mp4"],
+                ["examples/cinema_0.mp4"],
+                ["examples/cinema_1.mp4"],
+            ],
+            inputs=video_input,
+            label="Click on any example to load it"
+        )
+    
     with gr.Row():
         with gr.Column(scale=1):
-            # Video upload section
-            with gr.Group():
-                gr.Markdown("### 📹 Video Upload")
-                video_input = gr.Video(
-                    label="Upload Video",
-                    format="mp4"
-                )
-                
             # Interactive frame display
             with gr.Group():
                 gr.Markdown("### 🎯 Point Selection")
@@ -569,8 +581,9 @@ with gr.Blocks(
                     value="<p>Upload a video and select points to see 3D visualization here.</p>"
                 )
     
-    # Advanced settings section
-    with gr.Accordion("⚙️ Advanced Settings", open=False):
+    # Advanced settings section - changed to open=True
+    with gr.Accordion("⚙️ Advanced Settings", open=True):
+        gr.Markdown("Adjust these parameters to optimize tracking performance:")
         with gr.Row():
             grid_size = gr.Slider(
                 minimum=10,
@@ -578,7 +591,7 @@ with gr.Blocks(
                 step=10,
                 value=50,
                 label="Grid Size",
-                info="Size of the tracking grid"
+                info="Size of the tracking grid (larger = more detailed)"
             )
             vo_points = gr.Slider(
                 minimum=100,
@@ -586,7 +599,7 @@ with gr.Blocks(
                 step=50,
                 value=756,
                 label="VO Points",
-                info="Number of visual odometry points"
+                info="Number of visual odometry points (more = better accuracy)"
             )
             fps = gr.Slider(
                 minimum=1,
@@ -594,27 +607,13 @@ with gr.Blocks(
                 step=1,
                 value=3,
                 label="FPS",
-                info="Frames per second for processing"
+                info="Frames per second for processing (higher = smoother but slower)"
             )
     
     # Launch button
     with gr.Row():
         launch_btn = gr.Button("🚀 Launch Visualization", variant="primary", size="lg")
     
-    # Example videos section
-    with gr.Accordion("📂 Example Videos", open=False):
-        gr.Examples(
-            examples=[
-                ["examples/blackswan.mp4"],
-                ["examples/bike-packing.mp4"],
-                ["examples/bmx-trees.mp4"],
-                ["examples/breakdance.mp4"],
-                ["examples/camel.mp4"],
-            ],
-            inputs=video_input,
-            label="Try these example videos"
-        )
-    
     # Hidden state variables
     original_image_state = gr.State(None)
     selected_points = gr.State([])