Browse Source
l
Signed-off-by: xujinling <jinling.xu@zilliz.com>
main
xujinling
3 years ago
1 changed files with
4 additions and
4 deletions
-
get_configs.py
|
|
@ -3,14 +3,14 @@ |
|
|
def configs(model_name): |
|
|
def configs(model_name): |
|
|
args = { |
|
|
args = { |
|
|
'clip_initialized_model': |
|
|
'clip_initialized_model': |
|
|
{"side_size": 224, |
|
|
|
|
|
"crop_size": 256, |
|
|
|
|
|
|
|
|
{"side_size": 256, |
|
|
|
|
|
"crop_size": 224, |
|
|
"num_frames": 8, |
|
|
"num_frames": 8, |
|
|
"mean": [0.48145466, 0.4578275, 0.40821073], |
|
|
"mean": [0.48145466, 0.4578275, 0.40821073], |
|
|
"std": [0.26862954, 0.26130258, 0.27577711]}, |
|
|
"std": [0.26862954, 0.26130258, 0.27577711]}, |
|
|
'frozen_model': |
|
|
'frozen_model': |
|
|
{"side_size": 224, |
|
|
|
|
|
"crop_size": 256, |
|
|
|
|
|
|
|
|
{"side_size": 256, |
|
|
|
|
|
"crop_size": 224, |
|
|
"num_frames": 4, |
|
|
"num_frames": 4, |
|
|
"mean": [0.485, 0.456, 0.406], |
|
|
"mean": [0.485, 0.456, 0.406], |
|
|
"std": [0.229, 0.224, 0.225], } |
|
|
"std": [0.229, 0.224, 0.225], } |
|
|
|
