如何在windows上跑通TP-GAN人脸转正

本次实验使用的 https://github.com/UnrealLink/TP-GAN 作者说这只是一堂CV课的演示代码，不是正规接口，没事有效果就行。

论文 《Beyond Face Rotation: Global and Local Perception GAN for Photorealistic and Identity Preserving Frontal View Synthesis》

 

操作系统windows 8.1，编译器VS2015

我们只测试不训练，所以只使用test.py，将test.py内容复制到工程主文件，同时将

这几个文件复制到工程路径下，然后下载测试模型，readme里面提供了google盘的路径。

根据import代码，将缺少的python库下好，torch，torchvision，numpy，matplotlib

接着，代码需要改改： test.py里面

parser.add_argument('-c', action='store_true', default=False, help='cpu only (no cuda)') False 改成 True

G.load_state_dict(torch.load(args.model) 改成 G.load_state_dict(torch.load(args.model, map_location='cpu'))

最后我的工程长这样：

 

测试阶段，直接跑就行。