Browse Source
l
Signed-off-by: xujinling <jinling.xu@zilliz.com>
main
xujinling
3 years ago
1 changed files with
2 additions and
2 deletions
-
video_swin_transformer.py
|
|
@ -23,7 +23,7 @@ class VideoSwinTransformer(NNOperator): |
|
|
Args: |
|
|
Args: |
|
|
model_name (`str`): |
|
|
model_name (`str`): |
|
|
Supported model names: |
|
|
Supported model names: |
|
|
- swin_tiny_patch244_window877_kinetics400_1k |
|
|
|
|
|
|
|
|
- swin_t_w877_k400_1k |
|
|
skip_preprocess (`str`): |
|
|
skip_preprocess (`str`): |
|
|
Flag to skip video transforms. |
|
|
Flag to skip video transforms. |
|
|
classmap (`str=None`): |
|
|
classmap (`str=None`): |
|
|
@ -32,7 +32,7 @@ class VideoSwinTransformer(NNOperator): |
|
|
The number of classification labels to be returned (ordered by possibility from high to low). |
|
|
The number of classification labels to be returned (ordered by possibility from high to low). |
|
|
""" |
|
|
""" |
|
|
def __init__(self, |
|
|
def __init__(self, |
|
|
model_name: str = 'swin_tiny_patch244_window877_kinetics400_1k', |
|
|
|
|
|
|
|
|
model_name: str = 'swin_t_w877_k400_1k', |
|
|
framework: str = 'pytorch', |
|
|
framework: str = 'pytorch', |
|
|
skip_preprocess: bool = False, |
|
|
skip_preprocess: bool = False, |
|
|
classmap: str = None, |
|
|
classmap: str = None, |
|
|
|
