Shortcuts

mmcls.models.MultiLabelLinearClsHead

class mmcls.models.MultiLabelLinearClsHead(num_classes, in_channels, loss={'loss_weight': 1.0, 'reduction': 'mean', 'type': 'CrossEntropyLoss', 'use_sigmoid': True}, init_cfg={'layer': 'Linear', 'std': 0.01, 'type': 'Normal'})[源代码]

Linear classification head for multilabel task.

参数
  • num_classes (int) – Number of categories.

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

  • loss (dict) – Config of classification loss.

  • init_cfg (dict | optional) – The extra init config of layers. Defaults to use dict(type=’Normal’, layer=’Linear’, std=0.01).

Read the Docs v: latest
Versions
master
latest
1.x
dev-1.x
Downloads
pdf
html
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.