wxywb
3 years ago
1 changed files with
25 additions and
3 deletions
-
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: |
|
|
|
|
