本项目提供年龄预测的深度学习模型PyTorch实现,支持人脸图像到对应年龄的直接映射。包含数据预处理及训练脚本。
脸部老化CAAE要求安装相关库:运行命令`-r requirements.txt` 安装依赖项,包括Torch Vision 0.4.0 和 Torch 1.2.0。使用Colab笔记本进行数据集的训练与推论(具体步骤待定)。可以通过快照方式获取代码,方法为git clone或下载此存储库的zip文件。
对于UTKFace的数据集,请先从已对齐和裁剪版本中安装相关要求并执行`main.py`。运行命令:`python main.py`
结果将基于UTKFace数据集中不同年龄段进行划分,并输出以下年龄区间的结果:
- 0~5,
- 5~10,
- 10~15,
- 16~20,
- 21~30,
- 31~40,
- 41~50,
- 51~60,
- 61~70,超过70
训练参数设置如下:`epoch:50, step:86 EG_L1_loss:0.075875 |G_img_loss:5.226651 G_tv_loss:0.003358 |Ez_loss:0.851948 D_img:0.998970 |D_reconst`