From e8c99dee5728532ff5721e71fad14e0416ee5bf3 Mon Sep 17 00:00:00 2001 From: zhang chen Date: Sat, 18 Dec 2021 16:19:33 +0800 Subject: [PATCH] matching embedding test output with towhee op output --- resnet_image_embedding.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/resnet_image_embedding.yaml b/resnet_image_embedding.yaml index ec63089..419b6fe 100644 --- a/resnet_image_embedding.yaml +++ b/resnet_image_embedding.yaml @@ -8,6 +8,6 @@ init: model_name: str call: input: - img_tensor: torch.Tensor + img_path: str output: feature_vector: numpy.ndarray