Browse Source
Fix image mode issue
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
1 changed files with
1 additions and
1 deletions
-
benchmark/run.py
|
|
|
@ -49,7 +49,7 @@ query_data = query_data[:query_size] if query_size else query_data |
|
|
|
# Warm up |
|
|
|
print('Warming up...') |
|
|
|
op = ops.image_embedding.timm(model_name=model_name, device=device).get_op() |
|
|
|
dim = towhee.glob('https://raw.githubusercontent.com/towhee-io/towhee/main/towhee_logo.png') \ |
|
|
|
dim = towhee.dc(['https://raw.githubusercontent.com/towhee-io/towhee/main/towhee_logo.png']) \ |
|
|
|
.image_decode() \ |
|
|
|
.image_embedding.timm(model_name=model_name, device=device)[0].shape[0] |
|
|
|
print(f'output dim: {dim}') |
|
|
|
|
