Skip to content
Snippets Groups Projects

ArcFace/SphereFace Loss

Merged Tiago de Freitas Pereira requested to merge arcface into master
1 file
+ 1
1
Compare changes
  • Side-by-side
  • Inline
@@ -166,7 +166,7 @@ class ModifiedLogitsHead(tf.keras.layers.Layer):
"""
Implements the modified logit from equation (5) of `SphereFace: Deep Hypersphere Embedding for Face Recognition <https://arxiv.org/pdf/1704.08063.pdf>`_
It basically transforms the regular logit function to :math:`||x_i||cos(\theta_{yi})`, where :math:`\theta_{yi}=||x_i||_2^2||W||_2^2`
It basically transforms the regular logit function to :math:`||x_i||cos(\\theta_{yi})`, where :math:`\\theta_{yi}=||x_i||_2^2||W||_2^2`
Parameters
----------
Loading