From 721fcc237b40f5b92f1d882d7e38922c52e983a3 Mon Sep 17 00:00:00 2001 From: Nikita Karaev Date: Thu, 28 Dec 2023 17:27:30 +0000 Subject: [PATCH] remove assert B==1 --- cotracker/models/evaluation_predictor.py | 1 - cotracker/predictor.py | 1 - 2 files changed, 2 deletions(-) diff --git a/cotracker/models/evaluation_predictor.py b/cotracker/models/evaluation_predictor.py index aa7308d..87f8e18 100644 --- 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) diff --git a/cotracker/predictor.py b/cotracker/predictor.py index 4122f0b..dc2cd96 100644 --- 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)