From d99e8aebd578b0bfa5097a0aeb9f4a2094a949f1 Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Wed, 14 Dec 2022 19:25:59 +0800 Subject: [PATCH] Fix image mode issue Signed-off-by: Jael Gu --- benchmark/run.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/benchmark/run.py b/benchmark/run.py index f3c71de..6672eee 100644 --- a/benchmark/run.py +++ b/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}')