minor

app.py

@@ -28,7 +28,7 @@ def check_ram_usage(threshold_percent=90):
 def main() -> None:
     # Load data once at startup using the function from vis_st4rtrack.py
     global global_data_cache
-    global_data_cache = load_trajectory_data(use_float16=True, max_frames=32)
+    global_data_cache = load_trajectory_data(use_float16=True, max_frames=100, traj_path="480p_cows", mask_folder="cows")
     
     app = fastapi.FastAPI()
     viser_manager = ViserProxyManager(app)

vis_st4rtrack.py

@@ -163,16 +163,16 @@ def load_trajectory_data(traj_path="results", use_float16=True, max_frames=None,
 def visualize_st4rtrack(
     traj_path: str = "results",
     up_dir: str = "-z", # should be +z or -z
-    max_frames: int = 32,
+    max_frames: int = 100,
     share: bool = False,
-    point_size: float = 0.0015,
+    point_size: float = 0.005,
     downsample_factor: int = 3,
     num_traj_points: int = 100,
     conf_thre_percentile: float = 1,
     traj_end_frame: int = 100,
     traj_start_frame: int = 0,
     traj_line_width: float = 3.,
-    fixed_length_traj: int = 15,
+    fixed_length_traj: int = 20,
     server: viser.ViserServer = None,
     use_float16: bool = True,
     preloaded_data: dict = None,  # Add this parameter to accept preloaded data
@@ -185,6 +185,7 @@ def visualize_st4rtrack(
     mid_anchor: bool = False,
     video_width: int = 320,   # Video display width
     video_height: int = 180,  # Video display height
+    camera_position: tuple[float, float, float] = (1e-3, 1.5, -0.2),
 ) -> None:
     log_memory_usage("at start of visualization")
     
@@ -195,7 +196,7 @@ def visualize_st4rtrack(
 
     @server.on_client_connect
     def _(client: viser.ClientHandle) -> None:
-        client.camera.position = (1e-3, 0.6, -0.08)
+        client.camera.position = camera_position
         client.camera.look_at = (0, 0, 0)
 
     # Configure the GUI panel size and layout