logo
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

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