FaceShifter —非官方的PyTorch实施
非官方实现借助。 在本文中,有两个用于全流水线的网络,即AEI-Net和HEAR-Net。 我们仅实现AEI-Net,这是用于人脸交换的主要网络。
数据集
准备资料
您需要下载并解压缩:
CelebA-HQ()
VGGFace()
预处理数据
预处理代码主要基于Nvidia的FFHQ预处理代码。 您可以使用多种处理功能修改我们的预处理程序,以更快地完成预处理步骤。
# build docker image from Dockerfile
docker build -t dlib:0.0 ./preprocess
# run docker container from image
docker run -itd --ipc host -v /PATH_TO_THIS_FOLDER/preprocess:/workspace -v
1