i

3 years ago · 3576f963f7
1 changed files with 94 additions and 0 deletions
--- a/style_transfer_animegan.yaml
+++ b/style_transfer_animegan.yaml
@ -0,0 +1,94 @@
 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:
            -s
                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'