siriusdemon / Build-Your-Own-Face-Model

Build your own ArcFace, CenterFace/Centernet
334 stars 69 forks source link

精度达不到24.pth #26

Open GGgrey opened 2 years ago

GGgrey commented 2 years ago

您好,我最近跑了一下您的代码,参数设置就是默认的coonfig.py中的参数,数据集用的清洗后的CASIA-WebFace,测试用的LFW中6000个样本对,但是精度达不到24.pth。

Test Model: /mnt/sunqiao/xyh/models/_0.pth Accuracy: 0.661 Threshold: 0.704

Test Model: /mnt/sunqiao/xyh/models/_10.pth Accuracy: 0.825 Threshold: 0.426

Test Model: /mnt/sunqiao/xyh/models/_20.pth Accuracy: 0.835 Threshold: 0.391

Test Model: /mnt/sunqiao/xyh/models/_30.pth Accuracy: 0.835 Threshold: 0.399

再往下训练也基本停在84%左右,达不到97%,不知道什么原因。

siriusdemon commented 2 years ago

我也很纳闷为什么同样的代码会有这么多的差异。你可以试试改改 batch size 或者初始化参数的方法。