Browse Source
Update
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
test
2 changed files with
4 additions and
2 deletions
-
README.md
-
__init__.py
|
|
@ -38,6 +38,8 @@ dc.glob['path']('./dog.jpg') \ |
|
|
.select('vec') \ |
|
|
.select('vec') \ |
|
|
.to_list() |
|
|
.to_list() |
|
|
``` |
|
|
``` |
|
|
|
|
|
[array([0. , 0. , 0. , ..., 0. , 0.01748613, |
|
|
|
|
|
0. ], dtype=float32)] |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
## Factory Constructor |
|
|
## Factory Constructor |
|
|
|
|
|
@ -15,5 +15,5 @@ |
|
|
from .timm_image import TimmImage |
|
|
from .timm_image import TimmImage |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def timm(model_name: str, num_classes: int = 1000, skip_preprocess: bool = False): |
|
|
|
|
|
return TimmImage(model_name=model_name, num_classes=num_classes, skip_preprocess=skip_preprocess) |
|
|
|
|
|
|
|
|
def timm(**kwargs): |
|
|
|
|
|
return TimmImage(kwargs) |
|
|
|
