本次实验使用的 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'))

最后我的工程长这样:

 

测试阶段,直接跑就行。

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐