代码地址 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') 来控制年龄,用自己的图片试试看。