备注

您正在阅读 MMClassification 0.x 版本的文档。MMClassification 0.x 会在 2022 年末被切换为次要分支。建议您升级到 MMClassification 1.0 版本，体验更多新特性和新功能。请查阅 MMClassification 1.0 的安装教程、迁移教程以及更新日志。

mmcls.models.ConformerHead¶

class mmcls.models.ConformerHead(num_classes, in_channels, init_cfg={'layer': 'Linear', 'std': 0.01, 'type': 'Normal'}, *args, **kwargs)[源代码]¶

Linear classifier head.

参数

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).