pytorch,face multi task,landmarks,age,gender
采用数据集:IMDB-WIKI ,其官方地址如下: https://data.vision.ee.ethz.ch/cvl/rrothe/imdb-wiki/
- 该项目使用的数据为WIKI的“faces only”数据。
- 且进行了一定的加工修改:
- 1)去掉了一些错误的年龄数据人脸数据;
- 2)添加了98个人脸关键点数据;
- 3)更改了数据标注格式,并保存为json文件,每个图片对应一个json标注文件。
该项目的数据链接地址为:百度网盘,提取码:vvyj
将数据压缩文件wiki_crop_face.zip在根目录进行解压。
该项目的数据链接地址为:百度网盘,提取码:yxq8
1、模型训练,根目录下运行:python train.py
2、模型前向,在inference.py脚本中设置模型路径后,在根目录下运行:python inference.py
- E-mails: 305141918@qq.com