LabelData — mmengine 0.10.7 documentation (original) (raw)

class mmengine.structures.LabelData(*, metainfo=None, **kwargs)[source]¶

Data structure for label-level annotations or predictions.

Parameters:

metainfo (dict | None) –

static label_to_onehot(label, num_classes)[source]¶

Convert the label-format input to one-hot.

Parameters:

• label (torch.Tensor) – The label-format input. The format of item must be label-format.
• num_classes (int) – The number of classes.

Returns:

The converted results.

Return type:

torch.Tensor

static onehot_to_label(onehot)[source]¶

Convert the one-hot input to label.

Parameters:

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

Returns:

The converted results.

Return type:

torch.Tensor