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.2 KiB
71 lines
1.2 KiB
name: image_embedding_resnet50
|
|
type: image-embedding
|
|
operators:
|
|
- name: _start_op
|
|
function: _start_op
|
|
init_args: null
|
|
inputs:
|
|
- 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: null
|
|
inputs:
|
|
- df: img_str
|
|
name: image_path
|
|
col: 0
|
|
outputs:
|
|
- df: image
|
|
iter_info:
|
|
type: map
|
|
- name: embedding_model
|
|
function: towhee/timm-image-embedding
|
|
tag: main
|
|
init_args:
|
|
model_name: resnet50
|
|
inputs:
|
|
- df: image
|
|
name: image
|
|
col: 0
|
|
outputs:
|
|
- df: embedding
|
|
iter_info:
|
|
type: map
|
|
- name: _end_op
|
|
function: _end_op
|
|
init_args: null
|
|
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: img_str
|
|
columns:
|
|
- name: img_path
|
|
vtype: str
|
|
- name: image
|
|
columns:
|
|
- name: image
|
|
vtype: towhee.types.Image
|
|
- name: embedding
|
|
columns:
|
|
- name: feature_vector
|
|
vtype: numpy.ndarray
|
|
- name: _end_df
|
|
columns:
|
|
- name: feature_vector
|
|
vtype: numpy.ndarray
|