N2V Manipulate Torch
N2V manipulation transform for PyTorch.
N2VManipulateTorch
Default augmentation for the N2V model.
This transform expects C(Z)YX dimensions.
Parameters:
| Name | Type | Description | Default |
|---|---|---|---|
n2v_manipulate_config
|
N2VManipulateConfig
|
N2V manipulation configuration. |
required |
device
|
str
|
The device on which operations take place, e.g. "cuda", "cpu" or "mps". |
None
|
Attributes:
| Name | Type | Description |
|---|---|---|
masked_pixel_percentage |
float
|
Percentage of pixels to mask. |
roi_size |
int
|
Size of the replacement area. |
strategy |
Literal[uniform, median]
|
Replacement strategy, uniform or median. |
remove_center |
bool
|
Whether to remove central pixel from patch. |
struct_mask |
Optional[StructMaskParameters]
|
StructN2V mask parameters. |
rng |
Generator
|
Random number generator. |