fix_dense_tracking

henry123-boy · Jun 4, 2024 · 4e777d1 · 4e777d1
1 parent a4bedc5
commit 4e777d1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/models/spatracker/predictor.py b/models/spatracker/predictor.py
@@ -102,6 +102,8 @@ def _compute_dense_tracks(
                 queries=grid_pts,
                 backward_tracking=backward_tracking,
                 wind_length=wind_length,
+                video_depth=video_depth,
+                depth_predictor=depth_predictor,
             )
             tracks = smart_cat(tracks, tracks_step, dim=2)
             visibilities = smart_cat(visibilities, visibilities_step, dim=2)