TTA
Test-time augmentations.
ImageRestorationTTA
Test-time augmentation for image restoration tasks.
The augmentation is performed using all 90 deg rotations and their flipped version, as well as the original image flipped.
Tensors should be of shape SC(Z)YX.
This transformation is used in the LightningModule in order to perform test-time augmentation.
backward(x)
Undo the test-time augmentation.
Parameters:
| Name | Type | Description | Default |
|---|---|---|---|
x
|
Any
|
List of augmented tensors of shape SC(Z)YX. |
required |
Returns:
| Type | Description |
|---|---|
Any
|
Original tensor. |
forward(input_tensor)
Apply test-time augmentation to the input tensor.
Parameters:
| Name | Type | Description | Default |
|---|---|---|---|
input_tensor
|
Tensor
|
Input tensor, shape SC(Z)YX. |
required |
Returns:
| Type | Description |
|---|---|
list of torch.Tensor
|
List of augmented tensors. |