项目结构
- datasets:训练集
- lfw:测试集01
- testdata_3:测试集03
- logs:训练模型
- model_data:测试集对应的.txt文件。通过lfw_pair.py,可以将lfw和testdata_3等测试集转换为对应的.txt文件格式,然后通过eval_LFW.py直接调用.txt文件即可。
训练模型
首先调用txt_annotation.py将datasets训练集转化为txt文件格式(cls_train.txt),然后调用train.py读取
cls_train.txt进行模型训练,训练好的模型会存放在logs文件夹下。logs/loss_2022_04_19_11_56_34/epoch_acc.txt中可以查看各个模型对应的准确率。
测试数据集
测试数据集主要分为两步,首先需要调用lfw_pair.py将测试集转化为txt文件格式(lfw_pair.txt),然后调用eval_LFW.py读取lfw_pair.txt计算准确率