LabelData¶
- class mmengine.structures.LabelData(*, metainfo=None, **kwargs)[源代码]¶
Data structure for label-level annotations or predictions.
- 参数:
metainfo (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.
- 返回类型:
- 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.
- 返回类型: