Compare commits

...
10 Commits
Author SHA1 Message Date
david dff277f3a9 Add DeepFaceLab submodule 2026-06-24 13:53:24 -04:00
david 7d23263fe9 Activate deepfacelab conda environment and add pretrain_Quick96 zip 2026-06-24 13:53:19 -04:00
Nicholas Georgescu 7187736a69 added gpu check and potential nvidia fix that works on some systems (#71) 2023-07-11 16:29:23 +03:00
Claudio Sánchez 8cda581d2c fix "no training data provided" (#69)
* Update 6_train_Quick96.sh

Fix "no training data provided"

* Update 6_train_SAEHD.sh

fix "no training data provided"
2023-01-24 13:27:17 +03:00
Maximilian Wolf 05482b3825 Create 6_export_SAEHD_as_dfm.sh and 6_export_AMP_as_dfm.sh (#68) 2022-11-23 23:38:55 +03:00
githubcatw 65ca4c8ba5 Add support for generic XSeg (#65)
* Add generic XSeg
2022-10-12 10:48:58 +03:00
Struchkov Mark 3f860b0be4 Update 3.1_denoise_data_dst_images.sh (#37) 2021-05-19 14:14:08 +03:00
Alex 797e9e026c While the conda does not have new CUDNN and cudatoolkit packages, we use packages for version 2.3. fix #24
Thx for your attention, added the readme and changed the python version fix #25

HAPPY NEW YEAR!!!!!!!
2020-12-31 15:18:00 +05:00
Alex 2d331d7f31 Update CUDNN and Cuda versions for TF 2.4 2020-12-26 14:32:50 +05:00
Deepak Mangla 6de7d2fa20 Fetch only latest revision of Repo. Full repo is too large. (#15) 2020-12-09 12:48:22 +05:00
14 changed files with 62 additions and 7 deletions
+3
View File
@@ -0,0 +1,3 @@
[submodule "DeepFaceLab"]
path = DeepFaceLab
url = https://github.com/iperov/DeepFaceLab.git
Submodule
+1
Submodule DeepFaceLab added at e4b7543ffa
+19 -2
View File
@@ -17,12 +17,24 @@ Check latest cudnn and cudatoolkit version for your GPU device.
```bash ```bash
conda create -n deepfacelab -c main python=3.7 cudnn=7.6.5 cudatoolkit=10.1.243 conda create -n deepfacelab -c main python=3.7 cudnn=7.6.5 cudatoolkit=10.1.243
conda activate deepfacelab conda activate deepfacelab
git clone https://github.com/nagadit/DeepFaceLab_Linux.git git clone --depth 1 https://github.com/nagadit/DeepFaceLab_Linux.git
cd DeepFaceLab_Linux cd DeepFaceLab_Linux
git clone https://github.com/iperov/DeepFaceLab.git git clone --depth 1 https://github.com/iperov/DeepFaceLab.git
python -m pip install -r ./DeepFaceLab/requirements-cuda.txt python -m pip install -r ./DeepFaceLab/requirements-cuda.txt
``` ```
you can confirm your gpu is working correctly by running the following code and seeing what messages pop up:
```bash
python -c "import tensorflow as tf;print(tf.__version__)"
```
If the scripts can't seem to access the GPU and you're having issues with cuda version mismatches when running `nvidia-smi`, they can sometimes be remedied by simply running
```bash
conda install tensorflow-gpu==2.4.1
```
## 4. Download Pretrain (optional) ## 4. Download Pretrain (optional)
Use script 4.1 from the scripts directory. Use script 4.1 from the scripts directory.
@@ -35,3 +47,8 @@ Or download manually
[Quick96](https://github.com/nagadit/DeepFaceLab_Linux/releases/download/1.0/pretrain_Quick96.zip) [Quick96](https://github.com/nagadit/DeepFaceLab_Linux/releases/download/1.0/pretrain_Quick96.zip)
## 5. Navigate to the scripts directory and begin using DeepFaceLab_Linux ᗡ: ## 5. Navigate to the scripts directory and begin using DeepFaceLab_Linux ᗡ:
Run all scripts with BASH shell
```bash
bash 1_clear_workspace.sh
```
etc
Binary file not shown.
+4
View File
@@ -1,4 +1,8 @@
#!/usr/bin/env bash #!/usr/bin/env bash
source ~/miniconda3/etc/profile.d/conda.sh
conda activate deepfacelab
source ~/miniconda3/etc/profile.d/conda.sh
conda activate deepfacelab
source env.sh source env.sh
rm -r "$DFL_WORKSPACE" rm -r "$DFL_WORKSPACE"
+1 -1
View File
@@ -2,5 +2,5 @@
source env.sh source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" videoed denoise-image-sequence \ $DFL_PYTHON "$DFL_SRC/main.py" videoed denoise-image-sequence \
--output-dir "$DFL_WORKSPACE/data_dst" --input-dir "$DFL_WORKSPACE/data_dst"
+6
View File
@@ -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
@@ -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"
@@ -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"
+6
View File
@@ -0,0 +1,6 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" exportdfm \
--model-dir "$DFL_WORKSPACE/model" \
--model AMP
+6
View File
@@ -0,0 +1,6 @@
#!/usr/bin/env bash
source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" exportdfm \
--model-dir "$DFL_WORKSPACE/model" \
--model SAEHD
+2 -2
View File
@@ -4,8 +4,8 @@ source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" train \ $DFL_PYTHON "$DFL_SRC/main.py" train \
--training-data-src-dir "$DFL_WORKSPACE/data_src/aligned" \ --training-data-src-dir "$DFL_WORKSPACE/data_src/aligned" \
--training-data-dst-dir "$DFL_WORKSPACE/data_dst/aligned" \ --training-data-dst-dir "$DFL_WORKSPACE/data_dst/aligned" \
--pretraining-data-dir "$DFL_ROOT/pretrain_CelebA" \ --pretraining-data-dir "$DFL_SRC/pretrain_CelebA" \
--pretrained-model-dir "$DFL_ROOT/pretrain_Quick96" \ --pretrained-model-dir "$DFL_SRC/pretrain_Quick96" \
--model-dir "$DFL_WORKSPACE/model" \ --model-dir "$DFL_WORKSPACE/model" \
--model Quick96 --model Quick96
+1 -1
View File
@@ -4,7 +4,7 @@ source env.sh
$DFL_PYTHON "$DFL_SRC/main.py" train \ $DFL_PYTHON "$DFL_SRC/main.py" train \
--training-data-src-dir "$DFL_WORKSPACE/data_src/aligned" \ --training-data-src-dir "$DFL_WORKSPACE/data_src/aligned" \
--training-data-dst-dir "$DFL_WORKSPACE/data_dst/aligned" \ --training-data-dst-dir "$DFL_WORKSPACE/data_dst/aligned" \
--pretraining-data-dir "$DFL_ROOT/pretrain_CelebA" \ --pretraining-data-dir "$DFL_SRC/pretrain_CelebA" \
--model-dir "$DFL_WORKSPACE/model" \ --model-dir "$DFL_WORKSPACE/model" \
--model SAEHD --model SAEHD
+1 -1
View File
@@ -1,7 +1,7 @@
#!/usr/bin/env bash #!/usr/bin/env bash
conda activate deepfacelab conda activate deepfacelab
cd .. cd ..
export DFL_PYTHON="python3.6" export DFL_PYTHON="python3.7"
export DFL_WORKSPACE="workspace/" export DFL_WORKSPACE="workspace/"
if [ ! -d "$DFL_WORKSPACE" ]; then if [ ! -d "$DFL_WORKSPACE" ]; then