Compare commits
10
Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
dff277f3a9 | ||
|
|
7d23263fe9 | ||
|
|
7187736a69 | ||
|
|
8cda581d2c | ||
|
|
05482b3825 | ||
|
|
65ca4c8ba5 | ||
|
|
3f860b0be4 | ||
|
|
797e9e026c | ||
|
|
2d331d7f31 | ||
|
|
6de7d2fa20 |
@@ -0,0 +1,3 @@
|
|||||||
|
[submodule "DeepFaceLab"]
|
||||||
|
path = DeepFaceLab
|
||||||
|
url = https://github.com/iperov/DeepFaceLab.git
|
||||||
Submodule
+1
Submodule DeepFaceLab added at e4b7543ffa
@@ -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.
@@ -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"
|
||||||
|
|||||||
@@ -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"
|
||||||
|
|
||||||
|
|||||||
@@ -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"
|
||||||
@@ -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
|
||||||
@@ -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
|
||||||
@@ -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
|
||||||
|
|
||||||
|
|||||||
@@ -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
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user