Browse Source
modify mean
Signed-off-by: xujinling <jinling.xu@zilliz.com>
main
xujinling
3 years ago
1 changed files with
2 additions and
2 deletions
-
frozen_in_time.py
|
|
@ -73,8 +73,8 @@ class FrozenInTime(NNOperator): |
|
|
|
|
|
|
|
self.tokenizer = AutoTokenizer.from_pretrained('distilbert-base-uncased', TOKENIZERS_PARALLELISM=False) |
|
|
|
self.transform_cfgs = get_configs( |
|
|
|
side_size=224, |
|
|
|
crop_size=256, |
|
|
|
side_size=256, |
|
|
|
crop_size=224, |
|
|
|
num_frames=self.num_frames, |
|
|
|
mean=[0.485, 0.456, 0.406], |
|
|
|
std=[0.229, 0.224, 0.225], |
|
|
|