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: