From 136bb9e859e47f441ac92963a31ca6c57a24f937 Mon Sep 17 00:00:00 2001 From: xujinling Date: Wed, 22 Jun 2022 16:26:04 +0800 Subject: [PATCH] l Signed-off-by: xujinling --- bridge_former.py | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) diff --git a/bridge_former.py b/bridge_former.py index 400e6eb..14fcedb 100644 --- a/bridge_former.py +++ b/bridge_former.py @@ -52,14 +52,8 @@ class BridgeFormer(NNOperator): model_name=self.model_name) self.tokenizer = AutoTokenizer.from_pretrained('distilbert-base-uncased', TOKENIZERS_PARALLELISM=False) - # self.transform_cfgs = configs(self.model_name) - self.transform_cfgs = get_configs( - side_size=256, - crop_size=224, - num_frames=None, - mean=[0.485, 0.456, 0.406], - std=[0.229, 0.224, 0.225], - ) + self.transform_cfgs = configs(self.model_name) + self.model.eval() def decoder_video(self, data: List[VideoFrame]): @@ -73,6 +67,7 @@ class BridgeFormer(NNOperator): video=video, **self.transform_cfgs ) + # [B x C x T x H x W] video = video.to(self.device)[None, ...] return video