Browse Source
modify readme
Signed-off-by: gexy5 <xinyu.ge@zilliz.com>
main
gexy5
2 years ago
2 changed files with
2 additions and
2 deletions
-
README.md
-
__init__.py
|
|
@ -64,7 +64,7 @@ model_name='omnivore_swinT', skip_preprocess=False, classmap=None, topk=5)*** |
|
|
|
|
|
|
|
***model_name***: *str* |
|
|
|
|
|
|
|
The name of pre-trained tsm model. |
|
|
|
The name of pre-trained uniformer model. |
|
|
|
|
|
|
|
Supported model names: |
|
|
|
- omnivore_swinT |
|
|
|
|
|
@ -16,4 +16,4 @@ from .omnivore import Omnivore |
|
|
|
|
|
|
|
|
|
|
|
def omnivore(**kwargs): |
|
|
|
return Omnivore(**kwargs) |
|
|
|
return Omnivore(**kwargs) |
|
|
|