本篇文章主要介绍如何让电脑预测年龄性别,具体实现代码与运行步骤请参考Github.
github上有详细的运行步骤,这里我介绍一下模型训练过程。
1、 数据提取
有很多与年龄性别相关的公开数据集,这里先根据原实验采用IMBD-wiki数据集,数据集来自IMDB-WIKI — 50多万张标注了年龄和性别的人脸图像集 。 在把图像数据输入模型之前,我们进行与前述预处理相同的步骤,即检测人脸区域并添加边距。
import os
list = os.listdir(Agepath)
for i in range(0, len(list)):
imgName = os.path.basename(list[i])
if os.path.splitext(imgName)[1] != ".jpg":
continue
if i % 50=