remove assert B==1

2023-12-28 17:27:30 +00:00
parent 9c9a97d158
commit 721fcc237b
2 changed files with 0 additions and 2 deletions
--- a/cotracker/models/evaluation_predictor.py
+++ b/cotracker/models/evaluation_predictor.py
@@ -38,7 +38,6 @@ class EvaluationPredictor(torch.nn.Module):
        B, N, D = queries.shape

        assert D == 3
-        assert B == 1

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

        video = video.reshape(B * T, C, H, W)
        video = F.interpolate(video, tuple(self.interp_shape), mode="bilinear", align_corners=True)