arcface-pytorch
arcface-pytorch copied to clipboard
nets/arcface embedding_size ,s分别代表什么
你好!我想更换一个新的backbone,但是这两变量不知道指什么,感谢!
=0 0啊?
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时,遇到了错误,希望能获得你的帮助,感谢!