原作者的requirements里tensorflow-gpu的库有点问题
我在cuda9.0,python3.6,tensorflow_gpu==1.12.0中能跑起gpu版本,其他的库直接安装-cuda就可以,deepfacelab中的步骤:
python main.py videoed extract-video --input-file ./data/src.mp4 --output-dir ./data/
python main.py extract --input-dir ./data/src --output-dir ./data/aligned_src/ --detectd --output-debug
python main.py videoed extract-video --input-file ./data/dst.mp4 --output-dir ./data/
python main.py extract --input-dir ./data/dst --output-dir ./data/aligned_dst/ --detector s3d --output-debug
python main.py train
python main.py merge
python main.py videoed video-from-sequence
详解一下里面部分参数:
我按照正常流程训练完,是有些问题的,这块一共是有5个过程,
videoed extract-video是将视频抽成图片,当然源域和目标域都需要抽取,
extract是从图片中抽出人脸,用s3fd模型,有几个参数,包括full face(从眉毛到下颚),whole face(全脸),
train是模型训练
merge是跑推理
video video-from-sequence是将推理出来的图片合成视频