diff --git a/style_transfer_animegan.yaml b/style_transfer_animegan.yaml new file mode 100644 index 0000000..3f40d67 --- /dev/null +++ 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' \ No newline at end of file