Browse Source
Update
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
test
2 changed files with
9 additions and
3 deletions
-
README.md
-
__init__.py
|
|
@ -52,11 +52,16 @@ Create the operator via the following factory method |
|
|
|
|
|
|
|
***model_name***: *str* |
|
|
|
|
|
|
|
The model name in string. |
|
|
|
If no model name is given, it will use the default value "resnet34". |
|
|
|
The model name in string. The default value is "resnet34". |
|
|
|
Refer [Timm Docs](https://fastai.github.io/timmdocs/#List-Models-with-Pretrained-Weights) to get a full list of supported models. |
|
|
|
|
|
|
|
skip_preprocess (bool): |
|
|
|
|
|
|
|
***num_classes***: *1000* |
|
|
|
|
|
|
|
The number of classes. The default value is 1000. |
|
|
|
It is related to model and dataset. |
|
|
|
|
|
|
|
***skip_preprocess***: *bool* |
|
|
|
Flag to control whether to skip image preprocess, defaults to False. |
|
|
|
If set to True, image preprocess steps such as transform, normalization will be skipped. |
|
|
|
In this case, the user should guarantee that all the input images are already reprocessed properly, and thus can be fed to model directly. |
|
|
|
|
|
@ -14,5 +14,6 @@ |
|
|
|
|
|
|
|
from .timm_image import TimmImage |
|
|
|
|
|
|
|
|
|
|
|
def timm(**kwargs): |
|
|
|
return TimmImage(**kwargs) |
|
|
|