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

fixed a small online processing bug

Browse Source
This commit is contained in:
Nikita Karaev 2024-01-05 14:55:54 +00:00
parent f084a93f28
commit fac27989b3
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
cotracker/predictor.py
View File

@ -201,6 +201,7 @@ class CoTrackerOnlinePredictor(torch.nn.Module):
grid_query_frame: int = 0, grid_query_frame: int = 0,
add_support_grid=False, add_support_grid=False,
): ):
B, T, C, H, W = video_chunk.shape
# Initialize online video processing and save queried points # Initialize online video processing and save queried points
# This needs to be done before processing *each new video* # This needs to be done before processing *each new video*
if is_first_step: if is_first_step:
@ -231,7 +232,7 @@ class CoTrackerOnlinePredictor(torch.nn.Module):
queries = torch.cat([queries, grid_pts], dim=1) queries = torch.cat([queries, grid_pts], dim=1)
self.queries = queries self.queries = queries
return (None, None) return (None, None)
B, T, C, H, W = video_chunk.shape
video_chunk = video_chunk.reshape(B * T, C, H, W) video_chunk = video_chunk.reshape(B * T, C, H, W)
video_chunk = F.interpolate( video_chunk = F.interpolate(
video_chunk, tuple(self.interp_shape), mode="bilinear", align_corners=True video_chunk, tuple(self.interp_shape), mode="bilinear", align_corners=True