Browse Source
Update README
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
test
1 changed files with
1 additions and
1 deletions
-
README.md
|
|
@ -14,7 +14,7 @@ An image embedding operator implemented with pretrained models provided by [Timm |
|
|
|
from towhee import ops |
|
|
|
import numpy as np |
|
|
|
|
|
|
|
img_encoder = ops.image_embedding.timm('resnet50') |
|
|
|
img_encoder = ops.image_embedding.timm(model_name='resnet50') |
|
|
|
fake_img = np.zeros((256, 256, 3)) |
|
|
|
image_embedding = img_encoder(fake_img) |
|
|
|
``` |
|
|
|