Class SimpleClassificationDF

models/networks/arcface.py:195–208

@MODELS.register_module()
class SimpleClassificationDF(Module):
    def __init__(self, cfg: dict, **kwargs):
        super(SimpleClassificationDF, self).__init__()
        assert 'backbone' in cfg, 'Config for Backbones is mandatory!'
        assert 'head' in cfg, 'Config for Heads is mandatory!'
        
        self.backbone = BACKBONES.get(cfg.backbone.type)(**cfg.backbone)
        self.head = HEADS.get(cfg.head.type)(**cfg.head)
        self.model = Sequential(*[self.backbone,
                                  self.head])
        
    def forward(self, x):
        x = self.model(x)
        return x


##################################  MobileFaceNet #############################################################