arcface-pytorch icon indicating copy to clipboard operation
arcface-pytorch copied to clipboard

nets/arcface embedding_size ,s分别代表什么

Open Ledgero opened this issue 2 years ago • 2 comments

你好!我想更换一个新的backbone,但是这两变量不知道指什么,感谢!

Ledgero avatar Mar 25 '23 13:03 Ledgero

=0 0啊?

bubbliiiing avatar Mar 27 '23 17:03 bubbliiiing

class Arcface(nn.Module): def init(self, num_classes=None, backbone="mobilefacenet", pretrained=False, mode="train"): super(Arcface, self).init() if backbone=="mobilefacenet": embedding_size = 128 s = 32 self.arcface = get_mbf(embedding_size=embedding_size, pretrained=pretrained)

    elif backbone=="mobilenetv1":
        embedding_size  = 512
        s               = 64
        self.arcface    = get_mobilenet(dropout_keep_prob=0.5, embedding_size=embedding_size, pretrained=pretrained)
    else:
        raise ValueError('Unsupported backbone - `{}`, Use mobilefacenet, mobilenetv1.'.format(backbone))

    self.mode = mode
    if mode == "train":
        self.head = Arcface_Head(embedding_size=embedding_size, num_classes=num_classes, s=s)

这里提及的 s, embedding_size分别对应了什么?我不是很明白,修改成自己写的backbone时,遇到了错误,希望能获得你的帮助,感谢!

Ledgero avatar Mar 28 '23 02:03 Ledgero