备注

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

mmcls.models.VisionTransformerClsHead

class mmcls.models.VisionTransformerClsHead(num_classes, in_channels, hidden_dim=None, act_cfg={'type': 'Tanh'}, init_cfg={'layer': 'Linear', 'type': 'Constant', 'val': 0}, *args, **kwargs)

Vision Transformer classifier head.

参数

• num_classes (int) – Number of categories excluding the background category.

• in_channels (int) – Number of channels in the input feature map.

• hidden_dim (int) – Number of the dimensions for hidden layer. Defaults to None, which means no extra hidden layer.

• act_cfg (dict) – The activation config. Only available during pre-training. Defaults to dict(type='Tanh').

• init_cfg (dict) – The extra initialization configs. Defaults to dict(type='Constant', layer='Linear', val=0).