Commit a8d9f3ed authored by Anthony Larcher's avatar Anthony Larcher
Browse files

arcface

parent 57295377
......@@ -91,10 +91,10 @@ def l2_norm(input,axis=1):
return output
class Arcface(torch.nn.Module):
class ArcFace(torch.nn.Module):
# implementation of additive margin softmax loss in https://arxiv.org/abs/1801.05599
def __init__(self, embedding_size, classnum, s=64., m=0.5):
super(Arcface, self).__init__()
super(ArcFace, self).__init__()
self.classnum = classnum
self.kernel = Parameter(torch.Tensor(embedding_size, classnum))
# initial kernel
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment