如何在Windows上跑通HRFAE人脸年龄编辑

代码地址 https://github.com/InterDigitalInc/HRFAE

论文 《High Resolution Face Age Editing》

人脸年龄编辑就是输入一张人脸照片,生成任意岁数的这个人的照片。感谢该工作的研究人员们提供如此良心的开源代码。

操作系统 windows 8.1、编译器 VS2015

首先确定你的VS2015能正常使用python,使用PyCharm的同学也一样,确定python环境搭建正常。

新建python工程,关联python环境,根据作者提供的依赖库,对应我的python环境分别是:

我的

然后我们开始整代码,我们只看效果,所以只用test.py

将test.py的内容复制到python工程的主文件内,

parser.add_argument('--config', type=str, default='params', help='path to the config file.') 中params 改成 001

将以下文件拷贝到工程目录下

由于电脑是网吧机,所以我们只用cpu。

把 trainer.py 中的 state_dict = torch.load(checkpoint_path) 改成 state_dict = torch.load(checkpoint_path, map_location='cpu')

最后,我们下载测试需要的模型,复制 logs/001/download.sh 中的模型网址,下载后改名为checkpoint,并存放在该路径下。

我们工程路径长这样:

在测试图片放在test/input 文件夹内,运行代码,结果保存在test/output内。

修改 parser.add_argument('--target_age', type=int, default=65, help='Age transform target, interger value between 20 and 70') 来控制年龄,用自己的图片试试看。

猜你喜欢

转载自blog.csdn.net/XLcaoyi/article/details/108793500