logo
Browse Source

modify2

Signed-off-by: xujinling <jinling.xu@zilliz.com>
main
xujinling 2 years ago
parent
commit
f0c0d1c10b
  1. 12
      README.md

12
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*

Loading…
Cancel
Save