OS: Windows 10
顯卡: GTX 960
CUDA: 9.0
cuDNN: 7.6.4
Terminal: 以管理員權限執行cmd.exe
檢查自己顯卡相容性
安裝相容的CUDA Tookkit版本
安裝相容CUDA的cuDNN版本
安裝Anaconda
使用conda建立虛擬化的python執行環境,例如命名為
conda create -n faceswap python=3.6conda activate faceswap
初始化訓練環境
git clone --depth 1 https://github.com/deepfakes/faceswap.gitcd faceswappython setup.py
Nvidia的docker只支援linux,而我們使用的是windows,所以enable不了...
開始使用
python faceswap.py -h
python faceswap.py extract -i src\b.mp4 -o src/face_targetpython 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
移除python虛擬環境
conda env remove --name faceswap
從中移除CUDA、Anaconda