iicd/cotracker
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Nikita Karaev 2023-12-28 17:17:04 +00:00
parent 53682167fe
commit 9c9a97d158
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
cotracker/predictor.py
View File

@ -55,9 +55,7 @@ class CoTrackerPredictor(torch.nn.Module):
return tracks, visibilities return tracks, visibilities
def _compute_dense_tracks( def _compute_dense_tracks(self, video, grid_query_frame, grid_size=80, backward_tracking=False):
self, video, grid_query_frame, grid_size=150, backward_tracking=False
):
*_, H, W = video.shape *_, H, W = video.shape
grid_step = W // grid_size grid_step = W // grid_size
grid_width = W // grid_step grid_width = W // grid_step