Initial commit

scripts/0_setup.sh

@@ -0,0 +1,21 @@
+#!/usr/bin/env bash
+cd "$(dirname $0)/.."
+
+mkdir -p .dfl
+mkdir -p workspace
+
+if [ ! -d .dfl/DeepFaceLab ]; then
+  echo "Cloning DeepFaceLab"
+  cd .dfl
+  git clone "https://github.com/iperov/DeepFaceLab.git"
+fi
+
+if [ ! -d .dfl/env ]; then
+  virtualenv -p python3 .dfl/env
+fi
+
+source .dfl/env/bin/activate
+
+pip install -r requirements.txt
+
+echo "Done."

scripts/1_clean_workspace.sh

@@ -0,0 +1,7 @@
+#!/usr/bin/env bash
+cd "$(dirname $0)/.."
+source scripts/env.sh
+
+rm -fr "$WORKSPACE/data_src/*"
+rm -fr "$WORKSPACE/data_dst/*"
+rm -fr "$WORKSPACE/model/*"

scripts/2_extract_images_from_video_data_src.sh

@@ -0,0 +1,7 @@
+#!/usr/bin/env bash
+cd "$(dirname $0)/.."
+source scripts/env.sh
+
+python "$DFL_MAIN" videoed extract-video \
+  --input-file "$WORKSPACE/data_src.*" \
+  --output-dir "$WORKSPACE/data_src"

scripts/3_extract_images_from_video_data_dst.sh

@@ -0,0 +1,7 @@
+#!/usr/bin/env bash
+cd "$(dirname $0)/.."
+source scripts/env.sh
+
+python "$DFL_MAIN" videoed extract-video \
+  --input-file "$WORKSPACE/data_dst.*" \
+  --output-dir "$WORKSPACE/data_dst"

scripts/4.1_data_src_extract_faces_MANUAL.sh

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+cd "$(dirname $0)/.."
+source scripts/env.sh
+
+python "$DFL_MAIN" extract \
+  --input-dir "$WORKSPACE/data_src" \
+  --output-dir "$WORKSPACE/data_src/aligned" \
+  --detector manual

scripts/4.1_data_src_extract_faces_S3FD.sh

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+cd "$(dirname $0)/.."
+source scripts/env.sh
+
+python "$DFL_MAIN" extract \
+  --input-dir "$WORKSPACE/data_src" \
+  --output-dir "$WORKSPACE/data_src/aligned" \
+  --detector s3fd

scripts/4.1_data_src_extract_faces_S3FD_+_manual_fix.sh

@@ -0,0 +1,9 @@
+#!/usr/bin/env bash
+cd "$(dirname $0)/.."
+source scripts/env.sh
+
+python "$DFL_MAIN" extract \
+  --input-dir "$WORKSPACE/data_src" \
+  --output-dir "$WORKSPACE/data_src/aligned" \
+  --detector s3fd \
+  --manual-fix

scripts/4.2_data_src_sort.sh

@@ -0,0 +1,6 @@
+#!/usr/bin/env bash
+cd "$(dirname $0)/.."
+source scripts/env.sh
+
+python "$DFL_MAIN" sort \
+  --input-dir "$WORKSPACE/data_src/aligned"

scripts/4.2_data_src_util_faceset_enhance.sh

@@ -0,0 +1,6 @@
+#!/usr/bin/env bash
+cd "$(dirname $0)/.."
+source scripts/env.sh
+
+python "$DFL_MAIN" facesettool enhance \
+  --input-dir "$WORKSPACE/data_src/aligned"

scripts/5.1_data_dst_extract_faces_MANUAL.sh

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+cd "$(dirname $0)/.."
+source scripts/env.sh
+
+python "$DFL_MAIN" extract \
+  --input-dir "$WORKSPACE/data_dst" \
+  --output-dir "$WORKSPACE/data_dst/aligned" \
+  --detector manual

scripts/5.1_data_dst_extract_faces_S3FD.sh

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+cd "$(dirname $0)/.."
+source scripts/env.sh
+
+python "$DFL_MAIN" extract \
+  --input-dir "$WORKSPACE/data_dst" \
+  --output-dir "$WORKSPACE/data_dst/aligned" \
+  --detector s3fd

scripts/5.1_data_dst_extract_faces_S3FD_+_manual_fix.sh

@@ -0,0 +1,9 @@
+#!/usr/bin/env bash
+cd "$(dirname $0)/.."
+source scripts/env.sh
+
+python "$DFL_MAIN" extract \
+  --input-dir "$WORKSPACE/data_dst" \
+  --output-dir "$WORKSPACE/data_dst/aligned" \
+  --detector s3fd \
+  --manual-fix

scripts/5.2_data_dst_sort.sh

@@ -0,0 +1,6 @@
+#!/usr/bin/env bash
+cd "$(dirname $0)/.."
+source scripts/env.sh
+
+python "$DFL_MAIN" sort \
+  --input-dir "$WORKSPACE/data_dst/aligned"

scripts/5.2_data_dst_util_faceset_enhance.sh

@@ -0,0 +1,6 @@
+#!/usr/bin/env bash
+cd "$(dirname $0)/.."
+source scripts/env.sh
+
+python "$DFL_MAIN" facesettool enhance \
+  --input-dir "$WORKSPACE/data_dst/aligned"

scripts/6_train_Quick96.sh

@@ -0,0 +1,9 @@
+#!/usr/bin/env bash
+cd "$(dirname $0)/.."
+source scripts/env.sh
+
+python "$DFL_MAIN" train \
+  --training-data-src-dir "$WORKSPACE/data_src/aligned" \
+  --training-data-dst-dir "$WORKSPACE/data_dst/aligned" \
+  --model-dir "$WORKSPACE/model" \
+  --model Quick96

scripts/6_train_SAEHD.sh

@@ -0,0 +1,9 @@
+#!/usr/bin/env bash
+cd "$(dirname $0)/.."
+source scripts/env.sh
+
+python "$DFL_MAIN" train \
+  --training-data-src-dir "$WORKSPACE/data_src/aligned" \
+  --training-data-dst-dir "$WORKSPACE/data_dst/aligned" \
+  --model-dir "$WORKSPACE/model" \
+  --model SAEHD

scripts/7_merge_Quick96.sh

@@ -0,0 +1,11 @@
+#!/usr/bin/env bash
+cd "$(dirname $0)/.."
+source scripts/env.sh
+
+python "$DFL_MAIN" merge \
+  --input-dir "$WORKSPACE/data_dst" \
+  --output-dir "$WORKSPACE/data_dst/merged" \
+  --output-mask-dir "$WORKSPACE/data_dst/merged_mask" \
+  --aligned-dir "$WORKSPACE/data_dst/aligned" \
+  --model-dir "$WORKSPACE/model" \
+  --model Quick96

scripts/7_merge_SAEHD.sh

@@ -0,0 +1,11 @@
+#!/usr/bin/env bash
+cd "$(dirname $0)/.."
+source scripts/env.sh
+
+python "$DFL_MAIN" merge \
+  --input-dir "$WORKSPACE/data_dst" \
+  --output-dir "$WORKSPACE/data_dst/merged" \
+  --output-mask-dir "$WORKSPACE/data_dst/merged_mask" \
+  --aligned-dir "$WORKSPACE/data_dst/aligned" \
+  --model-dir "$WORKSPACE/model" \
+  --model SAEHD

scripts/8_merged_to_mov_lossless.sh

@@ -0,0 +1,10 @@
+#!/usr/bin/env bash
+cd "$(dirname $0)/.."
+source scripts/env.sh
+
+python "$DFL_MAIN" videoed video-from-sequence \
+  --input-dir "$WORKSPACE/data_dst/merged" \
+  --output-file "$WORKSPACE/result.mov" \
+  --reference-file "$WORKSPACE/data_dst.*" \
+  --include-audio \
+  --lossless

scripts/8_merged_to_mp4.sh

@@ -0,0 +1,9 @@
+#!/usr/bin/env bash
+cd "$(dirname $0)/.."
+source scripts/env.sh
+
+python "$DFL_MAIN" videoed video-from-sequence \
+  --input-dir "$WORKSPACE/data_dst/merged" \
+  --output-file "$WORKSPACE/result.mp4" \
+  --reference-file "$WORKSPACE/data_dst.*" \
+  --include-audio

scripts/8_merged_to_mp4_lossless.sh

@@ -0,0 +1,10 @@
+#!/usr/bin/env bash
+cd "$(dirname $0)/.."
+source scripts/env.sh
+
+python "$DFL_MAIN" videoed video-from-sequence \
+  --input-dir "$WORKSPACE/data_dst/merged" \
+  --output-file "$WORKSPACE/result.mp4" \
+  --reference-file "$WORKSPACE/data_dst.*" \
+  --include-audio \
+  --lossless

scripts/env.sh

@@ -0,0 +1,4 @@
+DFL_MAIN=.dfl/DeepFaceLab/main.py
+WORKSPACE=workspace
+
+source .dfl/env/bin/activate