From f0c0d1c10beac84161827e0522c58f00a818c6c0 Mon Sep 17 00:00:00 2001 From: xujinling Date: Tue, 14 Jun 2022 16:43:45 +0800 Subject: [PATCH] modify2 Signed-off-by: xujinling --- README.md | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/README.md b/README.md index 35e53d2..cf4d895 100644 --- a/README.md +++ b/README.md @@ -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*