From 72f7459dcbf9edc26c1f1f8fb4a637f188fd1463 Mon Sep 17 00:00:00 2001
From: zhang chen <chen.zhang@zilliz.com>
Date: Mon, 20 Dec 2021 14:59:58 +0800
Subject: [PATCH] change input data param

---
 README.md                     |  2 +-
 image_embedding_resnet50.yaml | 26 ++------------------------
 2 files changed, 3 insertions(+), 25 deletions(-)

diff --git a/README.md b/README.md
index dc2f44e..76b518a 100644
--- a/README.md
+++ b/README.md
@@ -21,7 +21,7 @@ The pipeline returns a tuple `Tuple[('feature_vector', numpy.ndarray)]` containi
 - feature_vector:
   - the embedding of input image
   - data type: `numpy.ndarray`
-
+  - shape: (2048,)
 ## How to use
 
 1. Install [Towhee](https://github.com/towhee-io/towhee)
diff --git a/image_embedding_resnet50.yaml b/image_embedding_resnet50.yaml
index c0b53a8..b8a6a42 100644
--- a/image_embedding_resnet50.yaml
+++ b/image_embedding_resnet50.yaml
@@ -14,22 +14,6 @@ operators:
                 df: 'image'
         iter_info:
             type: map
-    -
-        name: 'preprocessing'
-        function: 'towhee/transform-image' # default author is towhee
-        tag: 'main' # optional
-        init_args:
-            size: 256
-        inputs:
-            -
-                df: 'image'
-                name: 'img_path'
-                col: 0
-        outputs:
-            -
-                df: 'image_preproc'
-        iter_info:
-            type: map
     -
         name: 'embedding_model'
         function: 'towhee/resnet-image-embedding' # same as 'resnet50-image-embedding', default user is towhee
@@ -38,8 +22,8 @@ operators:
             model_name: 'resnet50'
         inputs:
             -
-                df: 'image_preproc'
-                name: 'img_tensor'
+                df: 'image'
+                name: 'img_path'
                 col: 0
         outputs:
             -
@@ -73,12 +57,6 @@ dataframes:
             -
                 name: 'img_path'
                 vtype: 'str'
-    -
-        name: 'image_preproc'
-        columns:
-            -
-                name: 'img_transformed'
-                vtype: 'torch.Tensor'
     -
         name: 'embedding'
         columns: