logo
Browse Source

Update

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
test
Jael Gu 2 years ago
parent
commit
63467c6fdd
  1. 2
      README.md
  2. 4
      __init__.py

2
README.md

@ -38,6 +38,8 @@ dc.glob['path']('./dog.jpg') \
.select('vec') \
.to_list()
```
[array([0. , 0. , 0. , ..., 0. , 0.01748613,
0. ], dtype=float32)]
## Factory Constructor

4
__init__.py

@ -15,5 +15,5 @@
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)

Loading…
Cancel
Save