towhee
/
image-embedding-resnet50
copied
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Readme
Files and versions
71 lines
1.6 KiB
71 lines
1.6 KiB
name: 'image-embedding-resnet'
|
|
operators:
|
|
-
|
|
name: '_start_op'
|
|
function: '_start_op'
|
|
init_args:
|
|
inputs:
|
|
-
|
|
df: '_start_df'
|
|
name: 'img_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
|
|
init_args:
|
|
model_name: 'resnet50'
|
|
inputs:
|
|
-
|
|
df: 'image'
|
|
name: 'img_path'
|
|
col: 0
|
|
outputs:
|
|
-
|
|
df: 'embedding'
|
|
iter_info:
|
|
type: map
|
|
-
|
|
name: '_end_op'
|
|
function: '_end_op'
|
|
init_args:
|
|
inputs:
|
|
-
|
|
df: 'embedding'
|
|
name: 'feature_vector'
|
|
col: 0
|
|
outputs:
|
|
-
|
|
df: '_end_df'
|
|
iter_info:
|
|
type: map
|
|
dataframes:
|
|
-
|
|
name: '_start_df'
|
|
columns:
|
|
-
|
|
name: 'img_path'
|
|
vtype: 'str'
|
|
-
|
|
name: 'image'
|
|
columns:
|
|
-
|
|
name: 'img_path'
|
|
vtype: 'str'
|
|
-
|
|
name: 'embedding'
|
|
columns:
|
|
-
|
|
name: 'feature_vector'
|
|
vtype: 'numpy.ndarray'
|
|
-
|
|
name: '_end_df'
|
|
columns:
|
|
-
|
|
name: 'feature_vector'
|
|
vtype: 'numpy.ndarray'
|