matching embedding test output with towhee op output

3 years ago · e8c99dee57
1 changed files with 1 additions and 1 deletions
--- 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