diff --git a/image_embedding_resnet50.yaml b/image_embedding_resnet50.yaml index cf2c60f..d201a81 100644 --- a/image_embedding_resnet50.yaml +++ b/image_embedding_resnet50.yaml @@ -10,21 +10,37 @@ operators: df: '_start_df' name: 'img_path' col: 0 + outputs: + - + df: 'img_str' + iter_info: + type: map + - + name: 'image_decoder' + function: 'towhee/image-decoder' + tag: 'main' + init_args: + inputs: + - + df: 'img_str' + name: 'image_path' + col: 0 outputs: - df: 'image' iter_info: type: map + - name: 'embedding_model' function: 'towhee/resnet-image-embedding' # same as 'resnet50-image-embedding', default user is towhee - tag: 'main' # optional + tag: 'main' init_args: model_name: 'resnet50' inputs: - df: 'image' - name: 'img_path' + name: 'image' col: 0 outputs: - @@ -53,11 +69,17 @@ dataframes: name: 'img_path' vtype: 'str' - - name: 'image' + name: 'img_str' columns: - name: 'img_path' vtype: 'str' + - + name: 'image' + columns: + - + name: 'image' + vtype: 'towhee.types.Image' - name: 'embedding' columns: