logo
Browse Source

modify supported model name

Signed-off-by: gexy5 <xinyu.ge@zilliz.com>
main
gexy5 2 years ago
parent
commit
a0287daacf
  1. 1
      README.md
  2. 1
      omnivore.py

1
README.md

@ -72,7 +72,6 @@ model_name='omnivore_swinT', skip_preprocess=False, classmap=None, topk=5)***
- omnivore_swinB - omnivore_swinB
- omnivore_swinB_in21k - omnivore_swinB_in21k
- omnivore_swinL_in21k - omnivore_swinL_in21k
- omnivore_swinB_epic
***skip_preprocess***: *bool* ***skip_preprocess***: *bool*

1
omnivore.py

@ -28,7 +28,6 @@ class Omnivore(NNOperator):
- omnivore_swinB - omnivore_swinB
- omnivore_swinB_in21k - omnivore_swinB_in21k
- omnivore_swinL_in21k - omnivore_swinL_in21k
- omnivore_swinB_epic
skip_preprocess (`str`): skip_preprocess (`str`):
Flag to skip video transforms. Flag to skip video transforms.
predict (`bool`): predict (`bool`):

Loading…
Cancel
Save