初試 Deepfake

Failed

環境設置

#

本文使用系統

#

OS: Windows 10
顯卡: GTX 960
CUDA: 9.0
cuDNN: 7.6.4
Terminal: 以管理員權限執行cmd.exe

安裝步驟

#

  1. 檢查自己顯卡相容性

    https://developer.nvidia.com/cuda-gpus#computeopen_in_new

  2. 安裝相容的CUDA Tookkit版本

    例如Windows 10 x64 CUDA Toolkit 9.0open_in_new

  3. 安裝相容CUDA的cuDNN版本

    例如cuDNN v7.6.4, for CUDA 9.0open_in_new

  4. 安裝Anaconda

  5. 使用conda建立虛擬化的python執行環境,例如命名為

    conda create -n faceswap python=3.6
    conda activate faceswap
  6. 初始化訓練環境

    git clone --depth 1 https://github.com/deepfakes/faceswap.git
    cd faceswap
    python setup.py

    Nvidia的docker只支援linux,而我們使用的是windows,所以enable不了...

  7. 開始使用

    python faceswap.py -h

Data Preparation

#

python faceswap.py extract -i src\b.mp4 -o src/face_target
python faceswap.py extract -i src\c.avi -o src/face_source

開始訓練模型

python faceswap.py train -A src/face_source -B src/face_target -m src/face_source_to_target_model/ -bs 32 --memory-saving-gradients --optimizer-savings

抽取影格

python tools.py effmpeg -i src\c.avi -o src\frames_raw -s 00:19:00 -e 00:20:00 --logfile tmp.log

環境清理

#

  1. 移除python虛擬環境

    conda env remove --name faceswap
  2. 中移除CUDA、Anaconda