Browse Source
l
Signed-off-by: xujinling <jinling.xu@zilliz.com>
main
xujinling
3 years ago
1 changed files with
3 additions and
8 deletions
-
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 |
|
|
|
|
