Spaces:
Sleeping
Sleeping
fix
Browse files
app.py
CHANGED
|
@@ -117,9 +117,6 @@ print("✅ Tracking models loaded successfully!")
|
|
| 117 |
|
| 118 |
gr.set_static_paths(paths=[Path.cwd().absolute()/"_viz"])
|
| 119 |
|
| 120 |
-
# --- YOUR ORIGINAL FUNCTIONS (generate_camera_trajectory, render_from_pointcloud, run_spatial_tracker) ---
|
| 121 |
-
# [Keeping these exactly as provided in your snippet]
|
| 122 |
-
|
| 123 |
|
| 124 |
def generate_camera_trajectory(num_frames: int, movement_type: str, base_intrinsics: np.ndarray, scene_scale: float = 1.0) -> tuple:
|
| 125 |
speed = scene_scale * 0.02
|
|
@@ -213,6 +210,8 @@ def render_from_pointcloud(rgb_frames, depth_frames, intrinsics, original_extrin
|
|
| 213 |
|
| 214 |
@spaces.GPU
|
| 215 |
def run_spatial_tracker(video_tensor):
|
|
|
|
|
|
|
| 216 |
if tracker_model.spatrack.base_model is None:
|
| 217 |
tracker_model.spatrack.base_model = vggt4track_model
|
| 218 |
|
|
|
|
| 117 |
|
| 118 |
gr.set_static_paths(paths=[Path.cwd().absolute()/"_viz"])
|
| 119 |
|
|
|
|
|
|
|
|
|
|
| 120 |
|
| 121 |
def generate_camera_trajectory(num_frames: int, movement_type: str, base_intrinsics: np.ndarray, scene_scale: float = 1.0) -> tuple:
|
| 122 |
speed = scene_scale * 0.02
|
|
|
|
| 210 |
|
| 211 |
@spaces.GPU
|
| 212 |
def run_spatial_tracker(video_tensor):
|
| 213 |
+
if not hasattr(vggt4track_model, "infer"):
|
| 214 |
+
vggt4track_model.infer = lambda x: vggt4track_model(x)
|
| 215 |
if tracker_model.spatrack.base_model is None:
|
| 216 |
tracker_model.spatrack.base_model = vggt4track_model
|
| 217 |
|