一、模型训练相关文件
- 数据文件:train.rec,train.idx,property,lfw.bin,cfp_fp.bin, agedb_30.bin;其中前三个为自己的数据集,用来训练,后三个为下载的数据集,用来验证。
- 训练配置文件:insightface/recogniton/config.py
- 模型训练文件:insightface/recogniton/train.py
- 模型训练参数控制脚本:run_train.sh
- 启动训练命令:nohup bash ./run_train.sh > train_history_mines 2>&1 &,其中train_history_mines为训练记录文件。
二、训练过程如下
1、下载预训练权重r100和r50,放到insightface/models路径下
地址:https://github.com/deepinsight/insightface/wiki/Model-Zoo
2、下载年龄性别模型,放到insightface/models路径下