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
93 lines
2.1 KiB
93 lines
2.1 KiB
name: 'image-embedding-resnet'
|
|
operators:
|
|
-
|
|
name: '_start_op'
|
|
function: '_start_op'
|
|
init_args:
|
|
inputs:
|
|
-
|
|
df: '_start_df'
|
|
name: 'img'
|
|
col: 0
|
|
outputs:
|
|
-
|
|
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'
|
|
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
|
|
tag: 'main' # optional
|
|
init_args:
|
|
model_name: 'resnet50'
|
|
inputs:
|
|
-
|
|
df: 'image_preproc'
|
|
name: 'img_tensor'
|
|
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'
|
|
vtype: 'PIL.Image'
|
|
-
|
|
name: 'image'
|
|
columns:
|
|
-
|
|
name: 'img'
|
|
vtype: 'PIL.Image'
|
|
-
|
|
name: 'image_preproc'
|
|
columns:
|
|
-
|
|
name: 'img_transformed'
|
|
vtype: 'torch.Tensor'
|
|
-
|
|
name: 'embedding'
|
|
columns:
|
|
-
|
|
name: 'feature_vector'
|
|
vtype: 'numpy.ndarray'
|
|
-
|
|
name: '_end_df'
|
|
columns:
|
|
-
|
|
name: 'feature_vector'
|
|
vtype: 'numpy.ndarray'
|