SequentialLabels
SequentialLabels
Bases: LabelTransform
Remap labels in a label map so they become consecutive.
For example, if a label map has labels (0, 3, 5), then this will apply
a RemapLabels transform with remapping={3: 1, 5: 2},
and therefore the output image will have labels (0, 1, 2).
Examples:
>>> import torch
>>> import torchio as tio
>>> def get_image(*labels):
... tensor = torch.as_tensor(labels).reshape(1, 1, 1, -1)
... image = tio.LabelMap(tensor=tensor)
... return image
...
>>> img_with_bg = get_image(0, 5, 10)
>>> transform = tio.SequentialLabels()
>>> transform(img_with_bg).data
tensor([[[[0, 1, 2]]]])
>>> img_without_bg = get_image(7, 11, 99)
>>> transform(img_without_bg).data
tensor([[[[0, 1, 2]]]])
Note
This transformation is always fully invertible .
Warning
The background is typically represented with the label 0. There
will be zeros in the output image even if they are none in the input.
Parameters:
| Name | Type | Description | Default |
|---|---|---|---|
masking_method
|
TypeMaskingMethod
|
See |
None
|
**kwargs
|
See |
{}
|
__call__(data)
Transform data and return a result of the same type.
Parameters:
| Name | Type | Description | Default |
|---|---|---|---|
data
|
TypeTransformInput
|
Instance of |
required |
to_hydra_config()
Return a dictionary representation of the transform for Hydra instantiation.