Adding Linux shell scripts /bin/bash

This commit is contained in:
Chelsea E. Manning 2021-02-24 04:32:37 -05:00 committed by dongyuezhao
parent 46fc2397c5
commit 0d6c69644f
33 changed files with 319 additions and 0 deletions

12
scripts/1_clear_workspace.sh Executable file
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"

View file

@ -0,0 +1,7 @@
#!/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"

View file

@ -0,0 +1,7 @@
#!/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"

6
scripts/4.2_data_src_sort.sh Executable file
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"

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

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"

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

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

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

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

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

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

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
scripts/5.2_data_dst_sort.sh Executable file
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"

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

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

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

View file

@ -0,0 +1,8 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" labelingtool edit_mask \
--input-dir "$DFL_WORKSPACE/data_dst/aligned" \
--confirmed-dir "$DFL_WORKSPACE/data_dst/aligned_confirmed" \
--skipped-dir "$DFL_WORKSPACE/data_dst/aligned_skipped"

View file

@ -0,0 +1,9 @@
#!/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 \
--output-debug

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" \
--detector manual \
--output-debug \
--manual-output-debug-fix \

View file

@ -0,0 +1,9 @@
#!/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 \
--output-debug

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" \
--output-debug \
--detector s3fd \
--manual-fix

11
scripts/6_train_Quick96.sh Executable file
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

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
scripts/6_train_SAEHD.sh Executable file
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

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
scripts/7_merge_Quick96.sh Executable file
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
scripts/7_merge_SAEHD.sh Executable file
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
scripts/8_merged_to_avi.sh Executable file
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

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
scripts/8_merged_to_mp4.sh Executable file
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

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

18
scripts/env.sh Executable file
View file

@ -0,0 +1,18 @@
#!/usr/bin/env bash
conda activate deepfacelab
export DFL_PYTHON="python3.7"
export DFL_WORKSPACE="$DEEPFACELAB_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="$DEEPFACELAB_PATH"
export DFL_SRC="$DEEPFACELAB_PATH"