i

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'