入手github超热门机器学习项目faceswap
安装环境(windows)
一般建议直接使用Docker 方式启动项目,这里由于开发机器太垃圾了,只在win下安装启动了一遍,显存不足,跑一会儿就挂了,这里记录下环境配置过程
Python 版本建议 3.6+,Python
Tensorflow版本建议(1.15.0),Tensorflow
N卡用户 CUDA(10.0),cudnn(7.6.4.38),win下安装包需要到N卡官网找,需要注册为开发者
CUDA
CUDNN其他Python包
python -m pip install --upgrade psutil pathlib numpy==1.17.4 opencv-python==4.1.2.30 scikit-image Pillow==6.2.1 toposort fastcluster matplotlib==3.1.1 ffmpy==0.2.2 git+https://github.com/deepfakes/nvidia-ml-py3.git Keras==2.2.4
启动项目初始化
python setup.py
win下启动UI
python faceswap.py gui
Linux 启动训练
python faceswap.py train -A res/images/lzll_dzq_a -B res/images/lzll_dzq_b -m res/model/lzll_dzq -t original -bs 64 -it 1000000 -g 1 -s 100 -ss 25000 -ps 50 -L INFO