From 935eabd18e4e9de809162b2004fedc171f79d8f3 Mon Sep 17 00:00:00 2001 From: xujinling Date: Wed, 15 Jun 2022 19:47:25 +0800 Subject: [PATCH] l Signed-off-by: xujinling --- video_swin_transformer.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/video_swin_transformer.py b/video_swin_transformer.py index 9a1c67f..b596a00 100644 --- a/video_swin_transformer.py +++ b/video_swin_transformer.py @@ -68,14 +68,18 @@ class VideoSwinTransformer(NNOperator): def decoder_video(self, data: List[VideoFrame]): video = numpy.stack([img.astype(numpy.float32) / 255. for img in data], axis=0) + print(video.shape) assert len(video.shape) == 4 video = video.transpose(3, 0, 1, 2) # twhc -> ctwh + print(video.shape) video = transform_video( video=video, **self.transform_cfgs ) + print(video.shape) # [B x C x T x H x W] video = video.to(self.device)[None, ...] + print(video.shape) return video def __call__(self, video: List[VideoFrame]):