image-embedding-resnet50/image_embedding_resnet50.yaml

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