|
|
@ -27,7 +27,7 @@ import towhee |
|
|
|
( |
|
|
|
towhee.glob('./archery.mp4') |
|
|
|
.video_decode.ffmpeg() |
|
|
|
.action_classification.video_swin_transformer(model_name='swin_tiny_patch244_window877_kinetics400_1k') |
|
|
|
.action_classification.video_swin_transformer(model_name='swin_t_w877_k400_1k') |
|
|
|
.show() |
|
|
|
) |
|
|
|
``` |
|
|
@ -43,7 +43,7 @@ import towhee |
|
|
|
( |
|
|
|
towhee.glob['path']('./archery.mp4') |
|
|
|
.video_decode.ffmpeg['path', 'frames']() |
|
|
|
.action_classification.video_swin_transformer['frames', ('labels', 'scores', 'features')](model_name='swin_tiny_patch244_window877_kinetics400_1k') |
|
|
|
.action_classification.video_swin_transformer['frames', ('labels', 'scores', 'features')](model_name='swin_t_w877_k400_1k') |
|
|
|
.select['path', 'labels', 'scores', 'features']() |
|
|
|
.show(formatter={'path': 'video_path'}) |
|
|
|
) |
|
|
@ -64,10 +64,10 @@ model_name='swin_tiny_patch244_window877_kinetics400_1k', skip_preprocess=False, |
|
|
|
***model_name***: *str* |
|
|
|
|
|
|
|
The name of pre-trained model. Supported model names: |
|
|
|
- swin_tiny_patch244_window877_kinetics400_1k |
|
|
|
- swin_base_patch244_window877_kinetics400_1k |
|
|
|
- swin_small_patch244_window877_kinetics400_1k |
|
|
|
- swin_base_patch244_window877_kinetics400_22k |
|
|
|
- swin_b_w877_k400_1k |
|
|
|
- swin_s_w877_k400_1k |
|
|
|
- swin_t_w877_k400_1k |
|
|
|
- swin_b_w877_k400_22k |
|
|
|
|
|
|
|
***skip_preprocess***: *bool* |
|
|
|
|
|
|
|