fix

2023-07-17 18:39:55 -07:00
parent 6880e31b5b
commit c6878420f5
2 changed files with 2 additions and 2 deletions
--- a/cotracker/predictor.py
+++ b/cotracker/predictor.py
@@ -65,7 +65,7 @@ class CoTrackerPredictor(torch.nn.Module):
        return tracks, visibilities

    def _compute_dense_tracks(
-        self, video, grid_query_frame, grid_size=50, backward_tracking=False
+        self, video, grid_query_frame, grid_size=30, backward_tracking=False
    ):
        *_, H, W = video.shape
        grid_step = W // grid_size