From 3576f963f7329b05f975ab3c7390003464620787 Mon Sep 17 00:00:00 2001
From: Filip Haltmayer <filip.haltmayer@zilliz.com>
Date: Wed, 26 Jan 2022 16:09:58 -0800
Subject: [PATCH] i

---
 style_transfer_animegan.yaml | 94 ++++++++++++++++++++++++++++++++++++
 1 file changed, 94 insertions(+)
 create mode 100644 style_transfer_animegan.yaml

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