style-transfer-animegan/style_transfer_animegan.yaml

name: 'style-transfer-animegan'
type: 'style-transfer'
operators:
    -
        name: '_start_op'
        function: '_start_op'
        init_args:
        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:
        inputs:
            -
                df: 'img_str'
                name: 'image_path'
                col: 0
        outputs:
            -
                df: 'image'
        iter_info:
            type: map

    -
        name: 'transfer_model'
        function: 'filip-halt/animegan-style-transfer'
        tag: 'main'
        init_args:
            model_name: 'paprika'
        inputs:
            -
                df: 'image'
                name: 'image'
                col: 0
        outputs:
            -
                df: 'new_image'
        iter_info:
            type: map
    -
        name: '_end_op'
        function: '_end_op'
        init_args:
        inputs:
            -
                df: 'new_image'
                name: 'styled_image'
                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: 'new_image'
        columns:
            -
                name: 'styled_image'
                vtype: 'numpy.ndarray'
    -
        name: '_end_df'
        columns:
            -
                name: 'styled_image'
                vtype: 'numpy.ndarray'