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

remove assert B==1

Browse Source
This commit is contained in:
Nikita Karaev
2023-12-28 17:27:30 +00:00
parent 9c9a97d158
commit 721fcc237b
2 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
cotracker/models/evaluation_predictor.py
View File

@@ -38,7 +38,6 @@ class EvaluationPredictor(torch.nn.Module):
B, N, D = queries.shape B, N, D = queries.shape
assert D == 3 assert D == 3
assert B == 1
video = video.reshape(B * T, C, H, W) video = video.reshape(B * T, C, H, W)
video = F.interpolate(video, tuple(self.interp_shape), mode="bilinear", align_corners=True) video = F.interpolate(video, tuple(self.interp_shape), mode="bilinear", align_corners=True)

1
cotracker/predictor.py
View File

@@ -92,7 +92,6 @@ class CoTrackerPredictor(torch.nn.Module):
backward_tracking=False, backward_tracking=False,
): ):
B, T, C, H, W = video.shape B, T, C, H, W = video.shape
assert B == 1
video = video.reshape(B * T, C, H, W) video = video.reshape(B * T, C, H, W)
video = F.interpolate(video, tuple(self.interp_shape), mode="bilinear", align_corners=True) video = F.interpolate(video, tuple(self.interp_shape), mode="bilinear", align_corners=True)