From 9c9a97d158115f89e367382f99178db8abd06cb4 Mon Sep 17 00:00:00 2001 From: Nikita Karaev Date: Thu, 28 Dec 2023 17:17:04 +0000 Subject: [PATCH] fix --- cotracker/predictor.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/cotracker/predictor.py b/cotracker/predictor.py index 7480820..4122f0b 100644 --- a/cotracker/predictor.py +++ b/cotracker/predictor.py @@ -55,9 +55,7 @@ class CoTrackerPredictor(torch.nn.Module): return tracks, visibilities - def _compute_dense_tracks( - self, video, grid_query_frame, grid_size=150, backward_tracking=False - ): + def _compute_dense_tracks(self, video, grid_query_frame, grid_size=80, backward_tracking=False): *_, H, W = video.shape grid_step = W // grid_size grid_width = W // grid_step