towhee
/
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.3 KiB
93 lines
2.3 KiB
name: 'image_embedding_resnet50'
|
|
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/image-transform-template' #your transform-image repo name
|
|
tag: 'main' #tag to the repo, default is 'main'
|
|
init_args:
|
|
size: #size of image, such as 256
|
|
inputs:
|
|
-
|
|
df: 'image'
|
|
name: 'img'
|
|
col: 0
|
|
outputs:
|
|
-
|
|
df: 'image_preproc'
|
|
iter_info:
|
|
type: map
|
|
-
|
|
name: embedding_model
|
|
function: 'towhee/image-embedding-operator-template' #your image-embedding repo name
|
|
tag: 'main' #tag to the repo, default is 'main'
|
|
init_args:
|
|
model_name: #model_name for image-embedding operator, such as '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'
|