add all scripts from original DFL

This commit is contained in:
Alex
2020-08-05 00:31:25 +03:00
parent 71ceb60719
commit 4bee8f184e
69 changed files with 404 additions and 29 deletions
+12
View File
@@ -0,0 +1,12 @@
#!/usr/bin/env bash
source env.sh
rm -r "$DFL_WORKSPACE"
mkdir "$DFL_WORKSPACE"
mkdir "$DFL_WORKSPACE/data_src"
mkdir "$DFL_WORKSPACE/data_src/aligned"
mkdir "$DFL_WORKSPACE/data_src/aligned_debug"
mkdir "$DFL_WORKSPACE/data_dst"
mkdir "$DFL_WORKSPACE/data_dst/aligned"
mkdir "$DFL_WORKSPACE/data_dst/aligned_debug"
mkdir "$DFL_WORKSPACE/model"
+6
View File
@@ -0,0 +1,6 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" videoed extract-video \
--input-file "$DFL_WORKSPACE/data_src.*" \
--output-dir "$DFL_WORKSPACE/data_src"
+6
View File
@@ -0,0 +1,6 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" videoed denoise-image-sequence \
--output-dir "$DFL_WORKSPACE/data_dst"
+8
View File
@@ -0,0 +1,8 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" videoed extract-video \
--input-file "$DFL_WORKSPACE/data_dst.*" \
--output-dir "$DFL_WORKSPACE/data_dst" \
--fps 0
+6
View File
@@ -0,0 +1,6 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" sort \
--input-dir "$DFL_WORKSPACE/data_src/aligned"
+7
View File
@@ -0,0 +1,7 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" util \
--input-dir "$DFL_WORKSPACE/data_src/aligned" \
--add-landmarks-debug-images
+6
View File
@@ -0,0 +1,6 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" facesettool enhance \
--input-dir "$DFL_WORKSPACE/data_src/aligned"
+7
View File
@@ -0,0 +1,7 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" util \
--input-dir "$DFL_WORKSPACE/data_src/aligned" \
--restore-faceset-metadata
+7
View File
@@ -0,0 +1,7 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" util \
--input-dir "$DFL_WORKSPACE/data_src/aligned" \
--save-faceset-metadata
+7
View File
@@ -0,0 +1,7 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" util \
--input-dir "$DFL_WORKSPACE/data_src/aligned" \
--pack-faceset
+7
View File
@@ -0,0 +1,7 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" util \
--input-dir "$DFL_WORKSPACE/data_src/aligned" \
--unpack-faceset
+7
View File
@@ -0,0 +1,7 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" util \
--input-dir "$DFL_WORKSPACE/data_src/aligned" \
--recover-original-aligned-filename
+8
View File
@@ -0,0 +1,8 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" extract \
--input-dir "$DFL_WORKSPACE/data_src" \
--output-dir "$DFL_WORKSPACE/data_src/aligned" \
--detector manual
+8
View File
@@ -0,0 +1,8 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" extract \
--input-dir "$DFL_WORKSPACE/data_src" \
--output-dir "$DFL_WORKSPACE/data_src/aligned" \
--detector s3fd
+6
View File
@@ -0,0 +1,6 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" sort \
--input-dir "$DFL_WORKSPACE/data_dst/aligned"
+7
View File
@@ -0,0 +1,7 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" util \
--input-dir "$DFL_WORKSPACE/data_dst/aligned" \
--pack-faceset
+7
View File
@@ -0,0 +1,7 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" util \
--input-dir "$DFL_WORKSPACE/data_dst/aligned" \
--unpack-faceset
+7
View File
@@ -0,0 +1,7 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" util \
--input-dir "$DFL_WORKSPACE/data_dst/aligned" \
--recover-original-aligned-filename
+6
View File
@@ -0,0 +1,6 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" xseg fetch \
--input-dir "$DFL_WORKSPACE/data_dst/aligned" \
--model-dir "$DFL_WORKSPACE/model"
+5
View File
@@ -0,0 +1,5 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" xseg editor \
--input-dir "$DFL_WORKSPACE/data_dst/aligned"
+5
View File
@@ -0,0 +1,5 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" xseg fetch \
--input-dir "$DFL_WORKSPACE/data_dst/aligned"
+5
View File
@@ -0,0 +1,5 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" xseg remove_labels \
--input-dir "$DFL_WORKSPACE/data_dst/aligned"
@@ -0,0 +1,5 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" xseg remove \
--input-dir "$DFL_WORKSPACE/data_dst/aligned"
+6
View File
@@ -0,0 +1,6 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" xseg fetch \
--input-dir "$DFL_WORKSPACE/data_src/aligned" \
--model-dir "$DFL_WORKSPACE/model"
+5
View File
@@ -0,0 +1,5 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" xseg fetch \
--input-dir "$DFL_WORKSPACE/data_src/aligned"
+5
View File
@@ -0,0 +1,5 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" xseg remove_labels \
--input-dir "$DFL_WORKSPACE/data_src/aligned"
@@ -0,0 +1,5 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" xseg remove \
--input-dir "$DFL_WORKSPACE/data_src/aligned"
+8
View File
@@ -0,0 +1,8 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" train \
--training-data-src-dir "$DFL_WORKSPACE/data_src/aligned" \
--training-data-dst-dir "$DFL_WORKSPACE/data_dst/aligned" \
--model-dir "$DFL_WORKSPACE/model" \
--model XSeg
+10
View File
@@ -0,0 +1,10 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" extract \
--input-dir "$DFL_WORKSPACE/data_dst" \
--output-dir "$DFL_WORKSPACE/data_dst/aligned" \
--detector manual \
--max-faces-from-image 0 \
--output-debug
@@ -0,0 +1,12 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" extract \
--input-dir "$DFL_WORKSPACE/data_dst" \
--output-dir "$DFL_WORKSPACE/data_dst/aligned" \
--detector manual \
--max-faces-from-image 0 \
--output-debug \
--manual-output-debug-fix \
+10
View File
@@ -0,0 +1,10 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" extract \
--input-dir "$DFL_WORKSPACE/data_dst" \
--output-dir "$DFL_WORKSPACE/data_dst/aligned" \
--detector s3fd \
--max-faces-from-image 0 \
--output-debug
+11
View File
@@ -0,0 +1,11 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" extract \
--input-dir "$DFL_WORKSPACE/data_dst" \
--output-dir "$DFL_WORKSPACE/data_dst/aligned" \
--output-debug \
--detector s3fd \
--max-faces-from-image 0 \
--manual-fix
+11
View File
@@ -0,0 +1,11 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" train \
--training-data-src-dir "$DFL_WORKSPACE/data_src/aligned" \
--training-data-dst-dir "$DFL_WORKSPACE/data_dst/aligned" \
--pretraining-data-dir "$DFL_ROOT/pretrain_CelebA" \
--pretrained-model-dir "$DFL_ROOT/pretrain_Quick96" \
--model-dir "$DFL_WORKSPACE/model" \
--model Quick96
+12
View File
@@ -0,0 +1,12 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" train \
--training-data-src-dir "$DFL_WORKSPACE/data_src/aligned" \
--training-data-dst-dir "$DFL_WORKSPACE/data_dst/aligned" \
--pretraining-data-dir "$DFL_SRC/pretrain_CelebA" \
--pretrained-model-dir "$DFL_SRC/pretrain_Quick96" \
--model-dir "$DFL_WORKSPACE/model" \
--model Quick96 \
--no-preview
+10
View File
@@ -0,0 +1,10 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" train \
--training-data-src-dir "$DFL_WORKSPACE/data_src/aligned" \
--training-data-dst-dir "$DFL_WORKSPACE/data_dst/aligned" \
--pretraining-data-dir "$DFL_ROOT/pretrain_CelebA" \
--model-dir "$DFL_WORKSPACE/model" \
--model SAEHD
+11
View File
@@ -0,0 +1,11 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" train \
--training-data-src-dir "$DFL_WORKSPACE/data_src/aligned" \
--training-data-dst-dir "$DFL_WORKSPACE/data_dst/aligned" \
--pretraining-data-dir "$DFL_SRC/pretrain_CelebA" \
--model-dir "$DFL_WORKSPACE/model" \
--model SAEHD \
--no-preview
+11
View File
@@ -0,0 +1,11 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" merge \
--input-dir "$DFL_WORKSPACE/data_dst" \
--output-dir "$DFL_WORKSPACE/data_dst/merged" \
--output-mask-dir "$DFL_WORKSPACE/data_dst/merged_mask" \
--aligned-dir "$DFL_WORKSPACE/data_dst/aligned" \
--model-dir "$DFL_WORKSPACE/model" \
--model Quick96
+11
View File
@@ -0,0 +1,11 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" merge \
--input-dir "$DFL_WORKSPACE/data_dst" \
--output-dir "$DFL_WORKSPACE/data_dst/merged" \
--output-mask-dir "$DFL_WORKSPACE/data_dst/merged_mask" \
--aligned-dir "$DFL_WORKSPACE/data_dst/aligned" \
--model-dir "$DFL_WORKSPACE/model" \
--model SAEHD
+16
View File
@@ -0,0 +1,16 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" videoed video-from-sequence \
--input-dir "$DFL_WORKSPACE/data_dst/merged" \
--output-file "$DFL_WORKSPACE/result.avi" \
--reference-file "$DFL_WORKSPACE/data_dst.*" \
--include-audio
$DFL_PYTHON "$DFL_SRC/main.py" videoed video-from-sequence \
--input-dir "$DFL_WORKSPACE/data_dst/merged_mask" \
--output-file "$DFL_WORKSPACE/result_mask.avi" \
--reference-file "$DFL_WORKSPACE/data_dst.*" \
--lossless
+17
View File
@@ -0,0 +1,17 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" videoed video-from-sequence \
--input-dir "$DFL_WORKSPACE/data_dst/merged" \
--output-file "$DFL_WORKSPACE/result.mov" \
--reference-file "$DFL_WORKSPACE/data_dst.*" \
--include-audio \
--lossless
$DFL_PYTHON "$DFL_SRC/main.py" videoed video-from-sequence \
--input-dir "$DFL_WORKSPACE/data_dst/merged_mask" \
--output-file "$DFL_WORKSPACE/result_mask.mov" \
--reference-file "$DFL_WORKSPACE/data_dst.*" \
--lossless
+15
View File
@@ -0,0 +1,15 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" videoed video-from-sequence \
--input-dir "$DFL_WORKSPACE/data_dst/merged" \
--output-file "$DFL_WORKSPACE/result.mp4" \
--reference-file "$DFL_WORKSPACE/data_dst.*" \
--include-audio
$DFL_PYTHON "$DFL_SRC/main.py" videoed video-from-sequence \
--input-dir "$DFL_WORKSPACE/data_dst/merged_mask" \
--output-file "$DFL_WORKSPACE/result_mask.mp4" \
--reference-file "$DFL_WORKSPACE/data_dst.*" \
--lossless
+17
View File
@@ -0,0 +1,17 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" videoed video-from-sequence \
--input-dir "$DFL_WORKSPACE/data_dst/merged" \
--output-file "$DFL_WORKSPACE/result.mp4" \
--reference-file "$DFL_WORKSPACE/data_dst.*" \
--include-audio \
--lossless
$DFL_PYTHON "$DFL_SRC/main.py" videoed video-from-sequence \
--input-dir "$DFL_WORKSPACE/data_dst/merged_mask" \
--output-file "$DFL_WORKSPACE/result_mask.mp4" \
--reference-file "$DFL_WORKSPACE/data_dst.*" \
--lossless
+17
View File
@@ -0,0 +1,17 @@
#!/usr/bin/env bash
export DFL_PYTHON="python3.6"
export DFL_WORKSPACE="workspace/"
if [ ! -d "$DFL_WORKSPACE" ]; then
mkdir "$DFL_WORKSPACE"
mkdir "$DFL_WORKSPACE/data_src"
mkdir "$DFL_WORKSPACE/data_src/aligned"
mkdir "$DFL_WORKSPACE/data_src/aligned_debug"
mkdir "$DFL_WORKSPACE/data_dst"
mkdir "$DFL_WORKSPACE/data_dst/aligned"
mkdir "$DFL_WORKSPACE/data_dst/aligned_debug"
mkdir "$DFL_WORKSPACE/model"
fi
export DFL_ROOT="./"
export DFL_SRC="./DeepFaceLab"