TreB1eN / InsightFace_Pytorch

Pytorch0.4.1 codes for InsightFace
MIT License
1.72k stars 418 forks source link

the implementation of arcface #151

Open Watebear opened 3 years ago

Watebear commented 3 years ago

In model.py, when implement the arcface head, the code calculate like: phi = cos_theta - m, but in the papar, it should be : phi = cos_theta+m. In face, that is AM-softmax.

lmagoncalo commented 3 years ago

Are you sure that you saw it right, because the line phi = cos_theta - m is the in Am_softmax class and not Arcface.