image-embedding-resnet50/image_embedding_resnet50.yaml

name: 'image-embedding-resnet50'
operators:
    -
        name: '_start_op'
        function: '_start_op'
        init_args:
        inputs:
            -
                df: '_start_df'
                name: 'img_tensor'
                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_tensor'
                col: 0
        outputs:
            -
                df: 'image_preproc'
        iter_info:
            type: map
    -
        name: 'embedding_model'
        function: 'towhee/resnet50-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: 'cnn'
                col: 0
        outputs:
            -
                df: '_end_df'
        iter_info:
            type: map
dataframes:
    -
        name: '_start_df'
        columns:
            -
                name: 'img_tensor'
                vtype: 'PIL.Image'
    -
        name: 'image'
        columns:
            -
                name: 'img_tensor'
                vtype: 'PIL.Image'
    -
        name: 'image_preproc'
        columns:
            -
                name: 'img_transformed'
                vtype: 'torch.Tensor'
    -
        name: 'embedding'
        columns:
            -
                name: 'cnn'
                vtype: 'numpy.ndarray'
    -
        name: '_end_df'
        columns:
            -
                name: 'cnn'
                vtype: 'numpy.ndarray'