modify mean

Signed-off-by: xujinling <jinling.xu@zilliz.com>
3 years ago · a2a38851c8
1 changed files with 2 additions and 2 deletions
--- a/frozen_in_time.py
+++ b/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],