mmcls.models.ConformerHead¶ class mmcls.models.ConformerHead(num_classes, in_channels, init_cfg={'layer': 'Linear', 'std': 0.01, 'type': 'Normal'}, *args, **kwargs)[source]¶ Linear classifier head. Parameters num_classes (int) – Number of categories excluding the background category. in_channels (int) – Number of channels in the input feature map. init_cfg (dict | optional) – The extra init config of layers. Defaults to use dict(type='Normal', layer='Linear', std=0.01).