From 09ac69ecb60af4cc2e9dea1ab9490f4d2af51874 Mon Sep 17 00:00:00 2001 From: Roooy Date: Thu, 12 Sep 2024 10:14:24 +0900 Subject: [PATCH] fix brackets missing by mistake --- sat/data_video.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sat/data_video.py b/sat/data_video.py index f3764c4..25d17ee 100644 --- a/sat/data_video.py +++ b/sat/data_video.py @@ -408,7 +408,7 @@ class SFTDataset(Dataset): num_frames = self.max_num_frames start = int(self.skip_frms_num) end = int(ori_vlen - self.skip_frms_num) - indices = np.arange(start, end, max((end - start) // num_frames), 1).astype(int) + indices = np.arange(start, end, max((end - start) // num_frames, 1)).astype(int) temp_frms = vr.get_batch(np.arange(start, end)) assert temp_frms is not None tensor_frms = (