Shortcuts

LabelData

class mmengine.structures.LabelData(*, metainfo=None, **kwargs)[源代码]

Data structure for label-level annotations or predictions.

参数

metainfo (Optional[dict]) –

返回类型

None

static label_to_onehot(label, num_classes)[源代码]

Convert the label-format input to one-hot.

参数
  • label (torch.Tensor) – The label-format input. The format of item must be label-format.

  • num_classes (int) – The number of classes.

返回

The converted results.

返回类型

torch.Tensor

static onehot_to_label(onehot)[源代码]

Convert the one-hot input to label.

参数

onehot (torch.Tensor, optional) – The one-hot input. The format of input must be one-hot.

返回

The converted results.

返回类型

torch.Tensor

Read the Docs v: latest
Versions
latest
stable
Downloads
html
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.