NVIDIA vid2vid论文复现

1.环境

pytorch1.8 +cuda111 ,numpy=1.16.2,和pytorch1.8对应的numpy1.17.0会报错 

certifi==2021.5.30
charset-normalizer==2.0.12
cycler==0.11.0
dataclasses==0.8
decorator==4.4.2
dominate==2.6.0
idna==3.3
imageio==2.15.0
install==1.3.5
kiwisolver==1.3.1
matplotlib==3.3.4
networkx==2.5.1
numpy==1.16.2
opencv-python==4.6.0.66
Pillow==8.4.0
pyparsing==3.0.9
python-dateutil==2.8.2
PyWavelets==1.1.1
requests==2.27.1
scikit-image==0.17.2
scipy==1.5.4
six==1.16.0
tifffile==2020.9.3
torch==1.8.0+cu111
torchaudio==0.8.0
torchvision==0.9.0+cu111
typing_extensions==4.1.1
urllib3==1.26.10

2.预先训练模型

下载的face_model.zip解压之后放在如下位置