From ed155dc37aa9c776d7dbe3b64ea217b97a497520 Mon Sep 17 00:00:00 2001 From: Luke Barr Date: Sun, 12 May 2019 18:56:47 -0500 Subject: [PATCH] Development reset --- README.md | 88 +++++++++++-------- scripts/1_clear_workspace.sh | 12 +++ scripts/2_extract_PNG_from_video_data_src.sh | 7 ++ scripts/3_extract_PNG_from_video_data_dst.sh | 7 ++ scripts/3_other_denoise_extracted_data_dst.sh | 6 ++ scripts/4.2.0_data_src_sort_by_black.sh | 7 ++ scripts/4.2.1_data_src_sort_by_blur.sh | 7 ++ ....2.2_data_src_sort_by_similar_histogram.sh | 7 ++ scripts/4.2.3_data_src_sort_by_one_face.sh | 7 ++ .../4.2.4_data_src_sort_by_dissimilar_face.sh | 7 ++ ...4_data_src_sort_by_dissimilar_histogram.sh | 7 ++ scripts/4.2.5_data_src_sort_by_face_pitch.sh | 7 ++ scripts/4.2.5_data_src_sort_by_face_yaw.sh | 7 ++ scripts/4.2.6_data_src_sort_by_final.sh | 7 ++ .../4.2_other_data_src_sort_by_brightness.sh | 7 ++ scripts/4.2_other_data_src_sort_by_hue.sh | 7 ++ ...ther_data_src_sort_by_original_filename.sh | 7 ++ ...ata_src_util_add_landmarks_debug_images.sh | 7 ++ scripts/4_data_src_extract_faces_MANUAL.sh | 10 +++ .../4_data_src_extract_faces_MT_all_GPU.sh | 9 ++ ...data_src_extract_faces_MT_all_GPU_debug.sh | 10 +++ .../4_data_src_extract_faces_MT_best_GPU.sh | 8 ++ .../4_data_src_extract_faces_S3FD_all_GPU.sh | 9 ++ ...ta_src_extract_faces_S3FD_all_GPU_debug.sh | 10 +++ .../4_data_src_extract_faces_S3FD_best_GPU.sh | 8 ++ scripts/5.2_data_dst_sort_by_black.sh | 6 ++ scripts/5.2_data_dst_sort_by_one_face.sh | 7 ++ .../5.2_data_dst_sort_by_similar_histogram.sh | 7 ++ scripts/5.3_data_dst_sort_by_blur.sh | 7 ++ ...3_data_dst_sort_by_dissimilar_histogram.sh | 7 ++ scripts/5.3_data_dst_sort_by_face_pitch.sh | 7 ++ scripts/5.3_data_dst_sort_by_face_yaw.sh | 7 ++ .../5.3_data_dst_sort_by_original_filename.sh | 7 ++ scripts/5_data_dst_extract_faces_MANUAL.sh | 9 ++ ...MANUAL_RE-EXTRACT_DELETED_RESULTS_DEBUG.sh | 11 +++ ...dst_extract_faces_MT_all_GPU+manual_fix.sh | 11 +++ .../5_data_dst_extract_faces_MT_all_GPU.sh | 9 ++ .../5_data_dst_extract_faces_MT_best_GPU.sh | 8 ++ ...t_extract_faces_S3FD_all_GPU+manual_fix.sh | 11 +++ .../5_data_dst_extract_faces_S3FD_all_GPU.sh | 10 +++ .../5_data_dst_extract_faces_S3FD_best_GPU.sh | 9 ++ scripts/6_train_DF.sh | 9 ++ scripts/6_train_DF_no_preview.sh | 10 +++ scripts/6_train_FANSegmentator.sh | 9 ++ scripts/6_train_FANSegmentator_no_preview.sh | 10 +++ scripts/6_train_H128.sh | 9 ++ scripts/6_train_H128_no_preview.sh | 10 +++ scripts/6_train_H64.sh | 9 ++ scripts/6_train_H64_no_preview.sh | 10 +++ scripts/6_train_LIAEF128.sh | 9 ++ scripts/6_train_LIAEF128_no_preview.sh | 10 +++ scripts/6_train_SAE.sh | 8 ++ scripts/6_train_SAE_no_preview.sh | 9 ++ scripts/7_convert_DF.sh | 10 +++ scripts/7_convert_DF_debug.sh | 11 +++ scripts/7_convert_H128.sh | 10 +++ scripts/7_convert_H128_debug.sh | 11 +++ scripts/7_convert_H64.sh | 10 +++ scripts/7_convert_H64_debug.sh | 11 +++ scripts/7_convert_LIAEF128.sh | 10 +++ scripts/7_convert_LIAEF128_debug.sh | 11 +++ scripts/7_convert_SAE.sh | 10 +++ scripts/7_convert_SAE_debug.sh | 11 +++ scripts/8_converted_to_avi.sh | 8 ++ scripts/8_converted_to_mov_lossless+alpha.sh | 9 ++ scripts/8_converted_to_mp4.sh | 8 ++ scripts/8_converted_to_mp4_lossless+alpha.sh | 9 ++ scripts/env.sh | 19 ++++ 68 files changed, 642 insertions(+), 36 deletions(-) create mode 100755 scripts/1_clear_workspace.sh create mode 100755 scripts/2_extract_PNG_from_video_data_src.sh create mode 100755 scripts/3_extract_PNG_from_video_data_dst.sh create mode 100755 scripts/3_other_denoise_extracted_data_dst.sh create mode 100755 scripts/4.2.0_data_src_sort_by_black.sh create mode 100755 scripts/4.2.1_data_src_sort_by_blur.sh create mode 100755 scripts/4.2.2_data_src_sort_by_similar_histogram.sh create mode 100755 scripts/4.2.3_data_src_sort_by_one_face.sh create mode 100755 scripts/4.2.4_data_src_sort_by_dissimilar_face.sh create mode 100755 scripts/4.2.4_data_src_sort_by_dissimilar_histogram.sh create mode 100755 scripts/4.2.5_data_src_sort_by_face_pitch.sh create mode 100755 scripts/4.2.5_data_src_sort_by_face_yaw.sh create mode 100755 scripts/4.2.6_data_src_sort_by_final.sh create mode 100755 scripts/4.2_other_data_src_sort_by_brightness.sh create mode 100755 scripts/4.2_other_data_src_sort_by_hue.sh create mode 100755 scripts/4.2_other_data_src_sort_by_original_filename.sh create mode 100755 scripts/4.2_other_data_src_util_add_landmarks_debug_images.sh create mode 100755 scripts/4_data_src_extract_faces_MANUAL.sh create mode 100755 scripts/4_data_src_extract_faces_MT_all_GPU.sh create mode 100755 scripts/4_data_src_extract_faces_MT_all_GPU_debug.sh create mode 100755 scripts/4_data_src_extract_faces_MT_best_GPU.sh create mode 100755 scripts/4_data_src_extract_faces_S3FD_all_GPU.sh create mode 100755 scripts/4_data_src_extract_faces_S3FD_all_GPU_debug.sh create mode 100755 scripts/4_data_src_extract_faces_S3FD_best_GPU.sh create mode 100755 scripts/5.2_data_dst_sort_by_black.sh create mode 100755 scripts/5.2_data_dst_sort_by_one_face.sh create mode 100755 scripts/5.2_data_dst_sort_by_similar_histogram.sh create mode 100755 scripts/5.3_data_dst_sort_by_blur.sh create mode 100755 scripts/5.3_data_dst_sort_by_dissimilar_histogram.sh create mode 100755 scripts/5.3_data_dst_sort_by_face_pitch.sh create mode 100755 scripts/5.3_data_dst_sort_by_face_yaw.sh create mode 100755 scripts/5.3_data_dst_sort_by_original_filename.sh create mode 100755 scripts/5_data_dst_extract_faces_MANUAL.sh create mode 100755 scripts/5_data_dst_extract_faces_MANUAL_RE-EXTRACT_DELETED_RESULTS_DEBUG.sh create mode 100755 scripts/5_data_dst_extract_faces_MT_all_GPU+manual_fix.sh create mode 100755 scripts/5_data_dst_extract_faces_MT_all_GPU.sh create mode 100755 scripts/5_data_dst_extract_faces_MT_best_GPU.sh create mode 100755 scripts/5_data_dst_extract_faces_S3FD_all_GPU+manual_fix.sh create mode 100755 scripts/5_data_dst_extract_faces_S3FD_all_GPU.sh create mode 100755 scripts/5_data_dst_extract_faces_S3FD_best_GPU.sh create mode 100755 scripts/6_train_DF.sh create mode 100755 scripts/6_train_DF_no_preview.sh create mode 100755 scripts/6_train_FANSegmentator.sh create mode 100755 scripts/6_train_FANSegmentator_no_preview.sh create mode 100755 scripts/6_train_H128.sh create mode 100755 scripts/6_train_H128_no_preview.sh create mode 100755 scripts/6_train_H64.sh create mode 100755 scripts/6_train_H64_no_preview.sh create mode 100755 scripts/6_train_LIAEF128.sh create mode 100755 scripts/6_train_LIAEF128_no_preview.sh create mode 100755 scripts/6_train_SAE.sh create mode 100755 scripts/6_train_SAE_no_preview.sh create mode 100755 scripts/7_convert_DF.sh create mode 100755 scripts/7_convert_DF_debug.sh create mode 100755 scripts/7_convert_H128.sh create mode 100755 scripts/7_convert_H128_debug.sh create mode 100755 scripts/7_convert_H64.sh create mode 100755 scripts/7_convert_H64_debug.sh create mode 100755 scripts/7_convert_LIAEF128.sh create mode 100755 scripts/7_convert_LIAEF128_debug.sh create mode 100755 scripts/7_convert_SAE.sh create mode 100755 scripts/7_convert_SAE_debug.sh create mode 100755 scripts/8_converted_to_avi.sh create mode 100755 scripts/8_converted_to_mov_lossless+alpha.sh create mode 100755 scripts/8_converted_to_mp4.sh create mode 100755 scripts/8_converted_to_mp4_lossless+alpha.sh create mode 100755 scripts/env.sh diff --git a/README.md b/README.md index 2dd541c..9aa9a8c 100644 --- a/README.md +++ b/README.md @@ -1,51 +1,67 @@ -![](doc/DFL_welcome.jpg) +# Installation for Ubuntu 16.04 -![](doc/logo_cuda.jpg)![](doc/logo_opencl.jpg)![](doc/logo_keras.jpg)![](doc/logo_tensorflow.jpg)![](doc/logo_plaidml.jpg) +An installation script has been created to automatically install all of the required dependencies for Ubuntu 16.04. Clone the repository and run ``ubuntu16.04-cuda9-installer.sh`` from the root directory of DeepFaceLab_Linux. -#deepfakes #faceswap #face-swap #deep-learning #deeplearning #deep-neural-networks #deepface #deep-face-swap #fakeapp #fake-app #neural-networks #neural-nets +# Installation for Ubuntu 18.04 -## **DeepFaceLab** is a tool that utilizes machine learning to replace faces in videos. +#### Add NVIDIA package repositories +```bash +wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.0.130-1_amd64.deb +sudo dpkg -i cuda-repo-ubuntu1804_10.0.130-1_amd64.deb +sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub +sudo apt-get update +``` -If you like this software, please consider a donation. +#### Install NVIDIA Video Driver +```bash +sudo apt-get install --no-install-recommends nvidia-driver-418 +``` +**Reboot your system.** -GOAL: next DeepFacelab update. +## Easy Method -[Donate via Yandex.Money](https://money.yandex.ru/to/41001142318065) +### Install Anaconda3 +Download the installer [https://www.anaconda.com/distribution/#linux](https://www.anaconda.com/distribution/#linux). Install Anaconda3 and choose the defaults. You will also need to add conda to your path so you can complete the final steps. -[Donate via Paypal](https://www.paypal.com/cgi-bin/webscr?cmd=_s-xclick&hosted_button_id=KK5ZCH4JXWMQS&source=url) +### Install DeepFaceLab -[Donate via Alipay](https://i.loli.net/2019/01/13/5c3ae3829809f.jpg) +```bash +conda create -y -n deepfacelab python=3.6.6 cudatoolkit=9.0 cudnn=7.3.1 +conda activate deepfacelab +git clone https://github.com/lbfs/DeepFaceLab_Linux.git +cd DeepFaceLab_Linux +python -m pip install -r requirements-cuda.txt +``` -bitcoin:31mPd6DxPCzbpCMZk4k1koWAbErSyqkAXr +## Harder Alternate Method -- ### [Gallery](doc/gallery/doc_gallery.md) +For this method, we will create an Ubuntu 16.04 container on your system. In order to do this, we will need to install and configure LXD. +```bash +sudo snap install lxd +sudo lxd init +sudo adduser "$USER" lxd +``` +**Reboot or logout so the new group membership can take effect** -- ### Manuals: +After you have finished installing and configuring lxd to your needs. We will now need to create the container. -[English (google translated)](doc/manual_en_google_translated.pdf) +```bash +echo "root:$UID:1" | sudo tee -a /etc/subuid /etc/subgid #Only run once and never again! +wget https://blog.simos.info/wp-content/uploads/2018/06/lxdguiprofile.txt #Thanks to Simos Xenitellis for his GUI LXC profile! +lxc profile create gui +cat lxdguiprofile.txt | lxc profile edit gui +lxc launch --profile default --profile gui ubuntu:16.04 deepfacelab +# Wait 30s so the environment can fully setup without issue. +# Logging in before the inital setup is done can cause problems. +# The next command will fix the DeepFaceLab GUI to allow it to show up correctly. +lxc exec deepfacelab -- sh -c "echo 'export QT_X11_NO_MITSHM=1' >> /home/ubuntu/.bashrc" +``` -[На русском](doc/manual_ru.pdf) +You can now access your container at any time with the following command +```bash +lxc exec deepfacelab -- su ubuntu +``` -- ### [Prebuilt windows app](doc/doc_prebuilt_windows_app.md) +While in the container, change to your home directory with ``cd ~\`` and then run the installation instructions for Ubuntu 16.04 and you will have created an identical environment. -- ### Forks - -[Google Colab fork](https://github.com/chervonij/DFL-Colab) by @chervonij - -[Linux fork](https://github.com/lbfs/DeepFaceLab_Linux) by @lbfs - may be outdated - -- ### [Ready to work facesets](doc/doc_ready_to_work_facesets.md) - -- ### [Build and repository info](doc/doc_build_and_repository_info.md) - -- ### Communication groups: - -(Chinese) QQ group 951138799 for ML/AI experts - -[deepfakes (Chinese)](https://deepfakescn.com) - -[deepfakes (Chinese) (outdated) ](https://deepfakes.com.cn/) - -[mrdeepfakes (English)](https://mrdeepfakes.com/forums/) - -[reddit (English)](https://www.reddit.com/r/GifFakes/new/) \ No newline at end of file +**WARNING: Make sure you install the same video driver in the container as installed in the host!** diff --git a/scripts/1_clear_workspace.sh b/scripts/1_clear_workspace.sh new file mode 100755 index 0000000..f1c5627 --- /dev/null +++ b/scripts/1_clear_workspace.sh @@ -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" diff --git a/scripts/2_extract_PNG_from_video_data_src.sh b/scripts/2_extract_PNG_from_video_data_src.sh new file mode 100755 index 0000000..8764cfd --- /dev/null +++ b/scripts/2_extract_PNG_from_video_data_src.sh @@ -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" + diff --git a/scripts/3_extract_PNG_from_video_data_dst.sh b/scripts/3_extract_PNG_from_video_data_dst.sh new file mode 100755 index 0000000..d9e489e --- /dev/null +++ b/scripts/3_extract_PNG_from_video_data_dst.sh @@ -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" + diff --git a/scripts/3_other_denoise_extracted_data_dst.sh b/scripts/3_other_denoise_extracted_data_dst.sh new file mode 100755 index 0000000..6264565 --- /dev/null +++ b/scripts/3_other_denoise_extracted_data_dst.sh @@ -0,0 +1,6 @@ +#!/usr/bin/env bash +source env.sh + +$DFL_PYTHON "$DFL_SRC/main.py" videoed denoise-image-sequence \ + --input-dir "$DFL_WORKSPACE/data_dst" + diff --git a/scripts/4.2.0_data_src_sort_by_black.sh b/scripts/4.2.0_data_src_sort_by_black.sh new file mode 100755 index 0000000..46f001c --- /dev/null +++ b/scripts/4.2.0_data_src_sort_by_black.sh @@ -0,0 +1,7 @@ +#!/usr/bin/env bash +source env.sh + +$DFL_PYTHON "$DFL_SRC/main.py" sort \ + --input-dir "$DFL_WORKSPACE/data_src/aligned" \ + --by black + diff --git a/scripts/4.2.1_data_src_sort_by_blur.sh b/scripts/4.2.1_data_src_sort_by_blur.sh new file mode 100755 index 0000000..e7e86d3 --- /dev/null +++ b/scripts/4.2.1_data_src_sort_by_blur.sh @@ -0,0 +1,7 @@ +#!/usr/bin/env bash +source env.sh + +$DFL_PYTHON "$DFL_SRC/main.py" sort \ + --input-dir "$DFL_WORKSPACE/data_src/aligned" \ + --by blur + diff --git a/scripts/4.2.2_data_src_sort_by_similar_histogram.sh b/scripts/4.2.2_data_src_sort_by_similar_histogram.sh new file mode 100755 index 0000000..59181ae --- /dev/null +++ b/scripts/4.2.2_data_src_sort_by_similar_histogram.sh @@ -0,0 +1,7 @@ +#!/usr/bin/env bash +source env.sh + +$DFL_PYTHON "$DFL_SRC/main.py" sort \ + --input-dir "$DFL_WORKSPACE/data_src/aligned" \ + --by hist + diff --git a/scripts/4.2.3_data_src_sort_by_one_face.sh b/scripts/4.2.3_data_src_sort_by_one_face.sh new file mode 100755 index 0000000..217f494 --- /dev/null +++ b/scripts/4.2.3_data_src_sort_by_one_face.sh @@ -0,0 +1,7 @@ +#!/usr/bin/env bash +source env.sh + +$DFL_PYTHON "$DFL_SRC/main.py" sort \ + --input-dir "$DFL_WORKSPACE/data_src/aligned" \ + --by oneface + diff --git a/scripts/4.2.4_data_src_sort_by_dissimilar_face.sh b/scripts/4.2.4_data_src_sort_by_dissimilar_face.sh new file mode 100755 index 0000000..718c067 --- /dev/null +++ b/scripts/4.2.4_data_src_sort_by_dissimilar_face.sh @@ -0,0 +1,7 @@ +#!/usr/bin/env bash +source env.sh + +$DFL_PYTHON "$DFL_SRC/main.py" sort \ + --input-dir "$DFL_WORKSPACE/data_src/aligned" \ + --by face-dissim + diff --git a/scripts/4.2.4_data_src_sort_by_dissimilar_histogram.sh b/scripts/4.2.4_data_src_sort_by_dissimilar_histogram.sh new file mode 100755 index 0000000..8540945 --- /dev/null +++ b/scripts/4.2.4_data_src_sort_by_dissimilar_histogram.sh @@ -0,0 +1,7 @@ +#!/usr/bin/env bash +source env.sh + +$DFL_PYTHON "$DFL_SRC/main.py" sort \ + --input-dir "$DFL_WORKSPACE/data_src/aligned" \ + --by hist-dissim + diff --git a/scripts/4.2.5_data_src_sort_by_face_pitch.sh b/scripts/4.2.5_data_src_sort_by_face_pitch.sh new file mode 100755 index 0000000..5a1fe6b --- /dev/null +++ b/scripts/4.2.5_data_src_sort_by_face_pitch.sh @@ -0,0 +1,7 @@ +#!/usr/bin/env bash +source env.sh + +$DFL_PYTHON "$DFL_SRC/main.py" sort \ + --input-dir "$DFL_WORKSPACE/data_src/aligned" \ + --by face-pitch + diff --git a/scripts/4.2.5_data_src_sort_by_face_yaw.sh b/scripts/4.2.5_data_src_sort_by_face_yaw.sh new file mode 100755 index 0000000..79617ae --- /dev/null +++ b/scripts/4.2.5_data_src_sort_by_face_yaw.sh @@ -0,0 +1,7 @@ +#!/usr/bin/env bash +source env.sh + +$DFL_PYTHON "$DFL_SRC/main.py" sort \ + --input-dir "$DFL_WORKSPACE/data_src/aligned" \ + --by face-yaw + diff --git a/scripts/4.2.6_data_src_sort_by_final.sh b/scripts/4.2.6_data_src_sort_by_final.sh new file mode 100755 index 0000000..5b79bbf --- /dev/null +++ b/scripts/4.2.6_data_src_sort_by_final.sh @@ -0,0 +1,7 @@ +#!/usr/bin/env bash +source env.sh + +$DFL_PYTHON "$DFL_SRC/main.py" sort \ + --input-dir "$DFL_WORKSPACE/data_src/aligned" \ + --by final + diff --git a/scripts/4.2_other_data_src_sort_by_brightness.sh b/scripts/4.2_other_data_src_sort_by_brightness.sh new file mode 100755 index 0000000..f40b9f9 --- /dev/null +++ b/scripts/4.2_other_data_src_sort_by_brightness.sh @@ -0,0 +1,7 @@ +#!/usr/bin/env bash +source env.sh + +$DFL_PYTHON "$DFL_SRC/main.py" sort \ + --input-dir "$DFL_WORKSPACE/data_src/aligned" \ + --by brightness + diff --git a/scripts/4.2_other_data_src_sort_by_hue.sh b/scripts/4.2_other_data_src_sort_by_hue.sh new file mode 100755 index 0000000..81aaba5 --- /dev/null +++ b/scripts/4.2_other_data_src_sort_by_hue.sh @@ -0,0 +1,7 @@ +#!/usr/bin/env bash +source env.sh + +$DFL_PYTHON "$DFL_SRC/main.py" sort \ + --input-dir "$DFL_WORKSPACE/data_src/aligned" \ + --by hue + diff --git a/scripts/4.2_other_data_src_sort_by_original_filename.sh b/scripts/4.2_other_data_src_sort_by_original_filename.sh new file mode 100755 index 0000000..8a95487 --- /dev/null +++ b/scripts/4.2_other_data_src_sort_by_original_filename.sh @@ -0,0 +1,7 @@ +#!/usr/bin/env bash +source env.sh + +$DFL_PYTHON "$DFL_SRC/main.py" sort \ + --input-dir "$DFL_WORKSPACE/data_src/aligned" \ + --by origname + diff --git a/scripts/4.2_other_data_src_util_add_landmarks_debug_images.sh b/scripts/4.2_other_data_src_util_add_landmarks_debug_images.sh new file mode 100755 index 0000000..a8f4fdc --- /dev/null +++ b/scripts/4.2_other_data_src_util_add_landmarks_debug_images.sh @@ -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 + diff --git a/scripts/4_data_src_extract_faces_MANUAL.sh b/scripts/4_data_src_extract_faces_MANUAL.sh new file mode 100755 index 0000000..9969dd0 --- /dev/null +++ b/scripts/4_data_src_extract_faces_MANUAL.sh @@ -0,0 +1,10 @@ +#!/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" \ + --multi-gpu \ + --detector manual \ + --debug-dir "$DFL_WORKSPACE/data_src/aligned_debug" + diff --git a/scripts/4_data_src_extract_faces_MT_all_GPU.sh b/scripts/4_data_src_extract_faces_MT_all_GPU.sh new file mode 100755 index 0000000..fa405e3 --- /dev/null +++ b/scripts/4_data_src_extract_faces_MT_all_GPU.sh @@ -0,0 +1,9 @@ +#!/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" \ + --multi-gpu \ + --detector mt + diff --git a/scripts/4_data_src_extract_faces_MT_all_GPU_debug.sh b/scripts/4_data_src_extract_faces_MT_all_GPU_debug.sh new file mode 100755 index 0000000..224544b --- /dev/null +++ b/scripts/4_data_src_extract_faces_MT_all_GPU_debug.sh @@ -0,0 +1,10 @@ +#!/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" \ + --multi-gpu \ + --detector mt \ + --debug-dir "$DFL_WORKSPACE/data_src/aligned_debug" + diff --git a/scripts/4_data_src_extract_faces_MT_best_GPU.sh b/scripts/4_data_src_extract_faces_MT_best_GPU.sh new file mode 100755 index 0000000..a960c2a --- /dev/null +++ b/scripts/4_data_src_extract_faces_MT_best_GPU.sh @@ -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 mt + diff --git a/scripts/4_data_src_extract_faces_S3FD_all_GPU.sh b/scripts/4_data_src_extract_faces_S3FD_all_GPU.sh new file mode 100755 index 0000000..2dd60a0 --- /dev/null +++ b/scripts/4_data_src_extract_faces_S3FD_all_GPU.sh @@ -0,0 +1,9 @@ +#!/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" \ + --multi-gpu \ + --detector s3fd + diff --git a/scripts/4_data_src_extract_faces_S3FD_all_GPU_debug.sh b/scripts/4_data_src_extract_faces_S3FD_all_GPU_debug.sh new file mode 100755 index 0000000..0efe138 --- /dev/null +++ b/scripts/4_data_src_extract_faces_S3FD_all_GPU_debug.sh @@ -0,0 +1,10 @@ +#!/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" \ + --multi-gpu \ + --detector s3fd \ + --debug-dir "$DFL_WORKSPACE/data_src/aligned_debug" + diff --git a/scripts/4_data_src_extract_faces_S3FD_best_GPU.sh b/scripts/4_data_src_extract_faces_S3FD_best_GPU.sh new file mode 100755 index 0000000..e54eb10 --- /dev/null +++ b/scripts/4_data_src_extract_faces_S3FD_best_GPU.sh @@ -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 + diff --git a/scripts/5.2_data_dst_sort_by_black.sh b/scripts/5.2_data_dst_sort_by_black.sh new file mode 100755 index 0000000..1af8596 --- /dev/null +++ b/scripts/5.2_data_dst_sort_by_black.sh @@ -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" \ + --by black diff --git a/scripts/5.2_data_dst_sort_by_one_face.sh b/scripts/5.2_data_dst_sort_by_one_face.sh new file mode 100755 index 0000000..1a50985 --- /dev/null +++ b/scripts/5.2_data_dst_sort_by_one_face.sh @@ -0,0 +1,7 @@ +#!/usr/bin/env bash +source env.sh + +$DFL_PYTHON "$DFL_SRC/main.py" sort \ + --input-dir "$DFL_WORKSPACE/data_dst/aligned" \ + --by oneface + diff --git a/scripts/5.2_data_dst_sort_by_similar_histogram.sh b/scripts/5.2_data_dst_sort_by_similar_histogram.sh new file mode 100755 index 0000000..bcc7ab9 --- /dev/null +++ b/scripts/5.2_data_dst_sort_by_similar_histogram.sh @@ -0,0 +1,7 @@ +#!/usr/bin/env bash +source env.sh + +$DFL_PYTHON "$DFL_SRC/main.py" sort \ + --input-dir "$DFL_WORKSPACE/data_dst/aligned" \ + --by hist + diff --git a/scripts/5.3_data_dst_sort_by_blur.sh b/scripts/5.3_data_dst_sort_by_blur.sh new file mode 100755 index 0000000..c4a947c --- /dev/null +++ b/scripts/5.3_data_dst_sort_by_blur.sh @@ -0,0 +1,7 @@ +#!/usr/bin/env bash +source env.sh + +$DFL_PYTHON "$DFL_SRC/main.py" sort \ + --input-dir "$DFL_WORKSPACE/data_dst/aligned" \ + --by blur + diff --git a/scripts/5.3_data_dst_sort_by_dissimilar_histogram.sh b/scripts/5.3_data_dst_sort_by_dissimilar_histogram.sh new file mode 100755 index 0000000..4339295 --- /dev/null +++ b/scripts/5.3_data_dst_sort_by_dissimilar_histogram.sh @@ -0,0 +1,7 @@ +#!/usr/bin/env bash +source env.sh + +$DFL_PYTHON "$DFL_SRC/main.py" sort \ + --input-dir "$DFL_WORKSPACE/data_dst/aligned" \ + --by hist-dissim + diff --git a/scripts/5.3_data_dst_sort_by_face_pitch.sh b/scripts/5.3_data_dst_sort_by_face_pitch.sh new file mode 100755 index 0000000..4ba3a30 --- /dev/null +++ b/scripts/5.3_data_dst_sort_by_face_pitch.sh @@ -0,0 +1,7 @@ +#!/usr/bin/env bash +source env.sh + +$DFL_PYTHON "$DFL_SRC/main.py" sort \ + --input-dir "$DFL_WORKSPACE/data_dst/aligned" \ + --by face-pitch + diff --git a/scripts/5.3_data_dst_sort_by_face_yaw.sh b/scripts/5.3_data_dst_sort_by_face_yaw.sh new file mode 100755 index 0000000..cc0afe1 --- /dev/null +++ b/scripts/5.3_data_dst_sort_by_face_yaw.sh @@ -0,0 +1,7 @@ +#!/usr/bin/env bash +source env.sh + +$DFL_PYTHON "$DFL_SRC/main.py" sort \ + --input-dir "$DFL_WORKSPACE/data_dst/aligned" \ + --by face-yaw + diff --git a/scripts/5.3_data_dst_sort_by_original_filename.sh b/scripts/5.3_data_dst_sort_by_original_filename.sh new file mode 100755 index 0000000..db88636 --- /dev/null +++ b/scripts/5.3_data_dst_sort_by_original_filename.sh @@ -0,0 +1,7 @@ +#!/usr/bin/env bash +source env.sh + +$DFL_PYTHON "$DFL_SRC/main.py" sort \ + --input-dir "$DFL_WORKSPACE/data_dst/aligned" \ + --by origname + diff --git a/scripts/5_data_dst_extract_faces_MANUAL.sh b/scripts/5_data_dst_extract_faces_MANUAL.sh new file mode 100755 index 0000000..fe8d02e --- /dev/null +++ b/scripts/5_data_dst_extract_faces_MANUAL.sh @@ -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" \ + --multi-gpu \ + --detector manual \ + --debug-dir "$DFL_WORKSPACE/data_dst/aligned_debug" diff --git a/scripts/5_data_dst_extract_faces_MANUAL_RE-EXTRACT_DELETED_RESULTS_DEBUG.sh b/scripts/5_data_dst_extract_faces_MANUAL_RE-EXTRACT_DELETED_RESULTS_DEBUG.sh new file mode 100755 index 0000000..c4e646e --- /dev/null +++ b/scripts/5_data_dst_extract_faces_MANUAL_RE-EXTRACT_DELETED_RESULTS_DEBUG.sh @@ -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" \ + --multi-gpu \ + --detector manual \ + --manual-output-debug-fix \ + --debug-dir "$DFL_WORKSPACE/data_dst/aligned_debug" + diff --git a/scripts/5_data_dst_extract_faces_MT_all_GPU+manual_fix.sh b/scripts/5_data_dst_extract_faces_MT_all_GPU+manual_fix.sh new file mode 100755 index 0000000..060a517 --- /dev/null +++ b/scripts/5_data_dst_extract_faces_MT_all_GPU+manual_fix.sh @@ -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" \ + --multi-gpu \ + --detector mt \ + --manual-fix \ + --debug-dir "$DFL_WORKSPACE/data_dst/aligned_debug" + diff --git a/scripts/5_data_dst_extract_faces_MT_all_GPU.sh b/scripts/5_data_dst_extract_faces_MT_all_GPU.sh new file mode 100755 index 0000000..d000906 --- /dev/null +++ b/scripts/5_data_dst_extract_faces_MT_all_GPU.sh @@ -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" \ + --multi-gpu \ + --detector mt \ + --debug-dir "$DFL_WORKSPACE/data_dst/aligned_debug" diff --git a/scripts/5_data_dst_extract_faces_MT_best_GPU.sh b/scripts/5_data_dst_extract_faces_MT_best_GPU.sh new file mode 100755 index 0000000..2fdf9ef --- /dev/null +++ b/scripts/5_data_dst_extract_faces_MT_best_GPU.sh @@ -0,0 +1,8 @@ +#!/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 mt \ + --debug-dir "$DFL_WORKSPACE/data_dst/aligned_debug" diff --git a/scripts/5_data_dst_extract_faces_S3FD_all_GPU+manual_fix.sh b/scripts/5_data_dst_extract_faces_S3FD_all_GPU+manual_fix.sh new file mode 100755 index 0000000..ad6578b --- /dev/null +++ b/scripts/5_data_dst_extract_faces_S3FD_all_GPU+manual_fix.sh @@ -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" \ + --multi-gpu \ + --detector s3fd \ + --manual-fix \ + --debug-dir "$DFL_WORKSPACE/data_dst/aligned_debug" + diff --git a/scripts/5_data_dst_extract_faces_S3FD_all_GPU.sh b/scripts/5_data_dst_extract_faces_S3FD_all_GPU.sh new file mode 100755 index 0000000..cddda0c --- /dev/null +++ b/scripts/5_data_dst_extract_faces_S3FD_all_GPU.sh @@ -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" \ + --multi-gpu \ + --detector s3fd \ + --debug-dir "$DFL_WORKSPACE/data_dst/aligned_debug" + diff --git a/scripts/5_data_dst_extract_faces_S3FD_best_GPU.sh b/scripts/5_data_dst_extract_faces_S3FD_best_GPU.sh new file mode 100755 index 0000000..87d284e --- /dev/null +++ b/scripts/5_data_dst_extract_faces_S3FD_best_GPU.sh @@ -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 \ + --debug-dir "$DFL_WORKSPACE/data_dst/aligned_debug" + diff --git a/scripts/6_train_DF.sh b/scripts/6_train_DF.sh new file mode 100755 index 0000000..34863c9 --- /dev/null +++ b/scripts/6_train_DF.sh @@ -0,0 +1,9 @@ +#!/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 DF + diff --git a/scripts/6_train_DF_no_preview.sh b/scripts/6_train_DF_no_preview.sh new file mode 100755 index 0000000..c97a452 --- /dev/null +++ b/scripts/6_train_DF_no_preview.sh @@ -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" \ + --model-dir "$DFL_WORKSPACE/model" \ + --model DF \ + --no-preview + diff --git a/scripts/6_train_FANSegmentator.sh b/scripts/6_train_FANSegmentator.sh new file mode 100755 index 0000000..19a0673 --- /dev/null +++ b/scripts/6_train_FANSegmentator.sh @@ -0,0 +1,9 @@ +#!/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 FANSegmentator + diff --git a/scripts/6_train_FANSegmentator_no_preview.sh b/scripts/6_train_FANSegmentator_no_preview.sh new file mode 100755 index 0000000..f931d8e --- /dev/null +++ b/scripts/6_train_FANSegmentator_no_preview.sh @@ -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" \ + --model-dir "$DFL_WORKSPACE/model" \ + --model FANSegmentator \ + --no-preview + diff --git a/scripts/6_train_H128.sh b/scripts/6_train_H128.sh new file mode 100755 index 0000000..bbcdd49 --- /dev/null +++ b/scripts/6_train_H128.sh @@ -0,0 +1,9 @@ +#!/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 H128 + diff --git a/scripts/6_train_H128_no_preview.sh b/scripts/6_train_H128_no_preview.sh new file mode 100755 index 0000000..dbe2563 --- /dev/null +++ b/scripts/6_train_H128_no_preview.sh @@ -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" \ + --model-dir "$DFL_WORKSPACE/model" \ + --model H128 \ + --no-preview + diff --git a/scripts/6_train_H64.sh b/scripts/6_train_H64.sh new file mode 100755 index 0000000..b002b0b --- /dev/null +++ b/scripts/6_train_H64.sh @@ -0,0 +1,9 @@ +#!/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 H64 + diff --git a/scripts/6_train_H64_no_preview.sh b/scripts/6_train_H64_no_preview.sh new file mode 100755 index 0000000..6a867c4 --- /dev/null +++ b/scripts/6_train_H64_no_preview.sh @@ -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" \ + --model-dir "$DFL_WORKSPACE/model" \ + --model H64 \ + --no-preview + diff --git a/scripts/6_train_LIAEF128.sh b/scripts/6_train_LIAEF128.sh new file mode 100755 index 0000000..74502b5 --- /dev/null +++ b/scripts/6_train_LIAEF128.sh @@ -0,0 +1,9 @@ +#!/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 LIAEF128 + diff --git a/scripts/6_train_LIAEF128_no_preview.sh b/scripts/6_train_LIAEF128_no_preview.sh new file mode 100755 index 0000000..27051ef --- /dev/null +++ b/scripts/6_train_LIAEF128_no_preview.sh @@ -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" \ + --model-dir "$DFL_WORKSPACE/model" \ + --model LIAEF128 \ + --no-preview + diff --git a/scripts/6_train_SAE.sh b/scripts/6_train_SAE.sh new file mode 100755 index 0000000..34cf93d --- /dev/null +++ b/scripts/6_train_SAE.sh @@ -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 SAE diff --git a/scripts/6_train_SAE_no_preview.sh b/scripts/6_train_SAE_no_preview.sh new file mode 100755 index 0000000..9dabc34 --- /dev/null +++ b/scripts/6_train_SAE_no_preview.sh @@ -0,0 +1,9 @@ +#!/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 SAE \ + --no-preview diff --git a/scripts/7_convert_DF.sh b/scripts/7_convert_DF.sh new file mode 100755 index 0000000..3aa7329 --- /dev/null +++ b/scripts/7_convert_DF.sh @@ -0,0 +1,10 @@ +#!/usr/bin/env bash +source env.sh + +$DFL_PYTHON "$DFL_SRC/main.py" convert \ + --input-dir "$DFL_WORKSPACE/data_dst" \ + --output-dir "$DFL_WORKSPACE/data_dst/merged" \ + --aligned-dir "$DFL_WORKSPACE/data_dst/aligned" \ + --model-dir "$DFL_WORKSPACE/model" \ + --model DF + diff --git a/scripts/7_convert_DF_debug.sh b/scripts/7_convert_DF_debug.sh new file mode 100755 index 0000000..af1b261 --- /dev/null +++ b/scripts/7_convert_DF_debug.sh @@ -0,0 +1,11 @@ +#!/usr/bin/env bash +source env.sh + +$DFL_PYTHON "$DFL_SRC/main.py" convert \ + --input-dir "$DFL_WORKSPACE/data_dst" \ + --output-dir "$DFL_WORKSPACE/data_dst/merged" \ + --aligned-dir "$DFL_WORKSPACE/data_dst/aligned" \ + --model-dir "$DFL_WORKSPACE/model" \ + --model DF \ + --debug + diff --git a/scripts/7_convert_H128.sh b/scripts/7_convert_H128.sh new file mode 100755 index 0000000..3ab1a5d --- /dev/null +++ b/scripts/7_convert_H128.sh @@ -0,0 +1,10 @@ +#!/usr/bin/env bash +source env.sh + +$DFL_PYTHON "$DFL_SRC/main.py" convert \ + --input-dir "$DFL_WORKSPACE/data_dst" \ + --output-dir "$DFL_WORKSPACE/data_dst/merged" \ + --aligned-dir "$DFL_WORKSPACE/data_dst/aligned" \ + --model-dir "$DFL_WORKSPACE/model" \ + --model H128 + diff --git a/scripts/7_convert_H128_debug.sh b/scripts/7_convert_H128_debug.sh new file mode 100755 index 0000000..713e6d0 --- /dev/null +++ b/scripts/7_convert_H128_debug.sh @@ -0,0 +1,11 @@ +#!/usr/bin/env bash +source env.sh + +$DFL_PYTHON "$DFL_SRC/main.py" convert \ + --input-dir "$DFL_WORKSPACE/data_dst" \ + --output-dir "$DFL_WORKSPACE/data_dst/merged" \ + --aligned-dir "$DFL_WORKSPACE/data_dst/aligned" \ + --model-dir "$DFL_WORKSPACE/model" \ + --model H128 \ + --debug + diff --git a/scripts/7_convert_H64.sh b/scripts/7_convert_H64.sh new file mode 100755 index 0000000..215e8b2 --- /dev/null +++ b/scripts/7_convert_H64.sh @@ -0,0 +1,10 @@ +#!/usr/bin/env bash +source env.sh + +$DFL_PYTHON "$DFL_SRC/main.py" convert \ + --input-dir "$DFL_WORKSPACE/data_dst" \ + --output-dir "$DFL_WORKSPACE/data_dst/merged" \ + --aligned-dir "$DFL_WORKSPACE/data_dst/aligned" \ + --model-dir "$DFL_WORKSPACE/model" \ + --model H64 \ + diff --git a/scripts/7_convert_H64_debug.sh b/scripts/7_convert_H64_debug.sh new file mode 100755 index 0000000..4d35d35 --- /dev/null +++ b/scripts/7_convert_H64_debug.sh @@ -0,0 +1,11 @@ +#!/usr/bin/env bash +source env.sh + +$DFL_PYTHON "$DFL_SRC/main.py" convert \ + --input-dir "$DFL_WORKSPACE/data_dst" \ + --output-dir "$DFL_WORKSPACE/data_dst/merged" \ + --aligned-dir "$DFL_WORKSPACE/data_dst/aligned" \ + --model-dir "$DFL_WORKSPACE/model" \ + --model H64 \ + --debug + diff --git a/scripts/7_convert_LIAEF128.sh b/scripts/7_convert_LIAEF128.sh new file mode 100755 index 0000000..cd64e1e --- /dev/null +++ b/scripts/7_convert_LIAEF128.sh @@ -0,0 +1,10 @@ +#!/usr/bin/env bash +source env.sh + +$DFL_PYTHON "$DFL_SRC/main.py" convert \ + --input-dir "$DFL_WORKSPACE/data_dst" \ + --output-dir "$DFL_WORKSPACE/data_dst/merged" \ + --aligned-dir "$DFL_WORKSPACE/data_dst/aligned" \ + --model-dir "$DFL_WORKSPACE/model" \ + --model LIAEF128 + diff --git a/scripts/7_convert_LIAEF128_debug.sh b/scripts/7_convert_LIAEF128_debug.sh new file mode 100755 index 0000000..ce09e60 --- /dev/null +++ b/scripts/7_convert_LIAEF128_debug.sh @@ -0,0 +1,11 @@ +#!/usr/bin/env bash +source env.sh + +$DFL_PYTHON "$DFL_SRC/main.py" convert \ + --input-dir "$DFL_WORKSPACE/data_dst" \ + --output-dir "$DFL_WORKSPACE/data_dst/merged" \ + --aligned-dir "$DFL_WORKSPACE/data_dst/aligned" \ + --model-dir "$DFL_WORKSPACE/model" \ + --model LIAEF128 \ + --debug + diff --git a/scripts/7_convert_SAE.sh b/scripts/7_convert_SAE.sh new file mode 100755 index 0000000..0193e54 --- /dev/null +++ b/scripts/7_convert_SAE.sh @@ -0,0 +1,10 @@ +#!/usr/bin/env bash +source env.sh + +$DFL_PYTHON "$DFL_SRC/main.py" convert \ + --input-dir "$DFL_WORKSPACE/data_dst" \ + --output-dir "$DFL_WORKSPACE/data_dst/merged" \ + --aligned-dir "$DFL_WORKSPACE/data_dst/aligned" \ + --model-dir "$DFL_WORKSPACE/model" \ + --model SAE + diff --git a/scripts/7_convert_SAE_debug.sh b/scripts/7_convert_SAE_debug.sh new file mode 100755 index 0000000..3711006 --- /dev/null +++ b/scripts/7_convert_SAE_debug.sh @@ -0,0 +1,11 @@ +#!/usr/bin/env bash +source env.sh + +$DFL_PYTHON "$DFL_SRC/main.py" convert \ + --input-dir "$DFL_WORKSPACE/data_dst" \ + --output-dir "$DFL_WORKSPACE/data_dst/merged" \ + --aligned-dir "$DFL_WORKSPACE/data_dst/aligned" \ + --model-dir "$DFL_WORKSPACE/model" \ + --model SAE \ + --debug + diff --git a/scripts/8_converted_to_avi.sh b/scripts/8_converted_to_avi.sh new file mode 100755 index 0000000..65235ed --- /dev/null +++ b/scripts/8_converted_to_avi.sh @@ -0,0 +1,8 @@ +#!/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.*" + diff --git a/scripts/8_converted_to_mov_lossless+alpha.sh b/scripts/8_converted_to_mov_lossless+alpha.sh new file mode 100755 index 0000000..19f0db0 --- /dev/null +++ b/scripts/8_converted_to_mov_lossless+alpha.sh @@ -0,0 +1,9 @@ +#!/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.*" \ + --lossless + diff --git a/scripts/8_converted_to_mp4.sh b/scripts/8_converted_to_mp4.sh new file mode 100755 index 0000000..957a17c --- /dev/null +++ b/scripts/8_converted_to_mp4.sh @@ -0,0 +1,8 @@ +#!/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.*" + diff --git a/scripts/8_converted_to_mp4_lossless+alpha.sh b/scripts/8_converted_to_mp4_lossless+alpha.sh new file mode 100755 index 0000000..7fa3eac --- /dev/null +++ b/scripts/8_converted_to_mp4_lossless+alpha.sh @@ -0,0 +1,9 @@ +#!/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.*" \ + --lossless + diff --git a/scripts/env.sh b/scripts/env.sh new file mode 100755 index 0000000..70d1d1a --- /dev/null +++ b/scripts/env.sh @@ -0,0 +1,19 @@ +source ~/anaconda3/etc/profile.d/conda.sh 2> /dev/null +conda activate deepfacelab 2> /dev/null +cd .. + +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_SRC="./"