From 65ca4c8ba5454cc05056a4b98d62260682bcae06 Mon Sep 17 00:00:00 2001
From: githubcatw <narek.email@gmail.com>
Date: Wed, 12 Oct 2022 11:48:58 +0400
Subject: [PATCH] Add support for generic XSeg (#65)

* Add generic XSeg
---
 scripts/4.1_download_XSeg_generic.sh        | 6 ++++++
 scripts/5_XSeg_generic_wf_data_dst_apply.sh | 6 ++++++
 scripts/5_XSeg_generic_wf_data_src_apply.sh | 6 ++++++
 3 files changed, 18 insertions(+)
 create mode 100644 scripts/4.1_download_XSeg_generic.sh
 create mode 100644 scripts/5_XSeg_generic_wf_data_dst_apply.sh
 create mode 100644 scripts/5_XSeg_generic_wf_data_src_apply.sh

diff --git a/scripts/4.1_download_XSeg_generic.sh b/scripts/4.1_download_XSeg_generic.sh
new file mode 100644
index 0000000..9ebd69c
--- /dev/null
+++ b/scripts/4.1_download_XSeg_generic.sh
@@ -0,0 +1,6 @@
+#!/usr/bin/env bash
+source env.sh
+
+wget https://github.com/githubcatw/DeepFaceLab_Linux/releases/download/xseg/model_generic_xseg.zip
+unzip -q model_generic_xseg.zip -d "$DFL_SRC"
+rm model_generic_xseg.zip
\ No newline at end of file
diff --git a/scripts/5_XSeg_generic_wf_data_dst_apply.sh b/scripts/5_XSeg_generic_wf_data_dst_apply.sh
new file mode 100644
index 0000000..af19830
--- /dev/null
+++ b/scripts/5_XSeg_generic_wf_data_dst_apply.sh
@@ -0,0 +1,6 @@
+#!/usr/bin/env bash
+source env.sh
+
+$DFL_PYTHON "$DFL_SRC/main.py" xseg apply \
+    --input-dir "$DFL_WORKSPACE/data_dst/aligned" \
+    --model-dir "$DFL_SRC/model_generic_xseg"
diff --git a/scripts/5_XSeg_generic_wf_data_src_apply.sh b/scripts/5_XSeg_generic_wf_data_src_apply.sh
new file mode 100644
index 0000000..34401e3
--- /dev/null
+++ b/scripts/5_XSeg_generic_wf_data_src_apply.sh
@@ -0,0 +1,6 @@
+#!/usr/bin/env bash
+source env.sh
+
+$DFL_PYTHON "$DFL_SRC/main.py" xseg apply \
+    --input-dir "$DFL_WORKSPACE/data_src/aligned" \
+    --model-dir "$DFL_SRC/model_generic_xseg"