N2V Manipulate
N2V manipulation transform.
N2VManipulate
Bases: Transform
Default augmentation for the N2V model.
This transform expects C(Z)YX dimensions.
Parameters:
-
roi_size(int, default:11) –Size of the replacement area, by default 11.
-
masked_pixel_percentage(float, default:0.2) –Percentage of pixels to mask, by default 0.2.
-
strategy(Literal['uniform', 'median'], default:value) –Replaccement strategy, uniform or median, by default uniform.
-
remove_center(bool, default:True) –Whether to remove central pixel from patch, by default True.
-
struct_mask_axis(Literal['horizontal', 'vertical', 'none'], default:'none') –StructN2V mask axis, by default "none".
-
struct_mask_span(int, default:5) –StructN2V mask span, by default 5.
-
seed(Optional[int], default:None) –Random seed, by default None.
Attributes:
-
masked_pixel_percentage(float) –Percentage of pixels to mask.
-
roi_size(int) –Size of the replacement area.
-
strategy(Literal['uniform', 'median']) –Replaccement 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.
__call__(patch, *args, **kwargs)
Apply the transform to the image.
Parameters:
-
patch(ndarray) –Image patch, 2D or 3D, shape C(Z)YX.
-
*args(Any, default:()) –Additional arguments, unused.
-
**kwargs(Any, default:{}) –Additional keyword arguments, unused.
Returns:
-
tuple[ndarray, ndarray, ndarray]–Masked patch, original patch, and mask.
__init__(roi_size=11, masked_pixel_percentage=0.2, strategy=SupportedPixelManipulation.UNIFORM.value, remove_center=True, struct_mask_axis='none', struct_mask_span=5, seed=None)
Constructor.
Parameters:
-
roi_size(int, default:11) –Size of the replacement area, by default 11.
-
masked_pixel_percentage(float, default:0.2) –Percentage of pixels to mask, by default 0.2.
-
strategy(Literal['uniform', 'median'], default:value) –Replaccement strategy, uniform or median, by default uniform.
-
remove_center(bool, default:True) –Whether to remove central pixel from patch, by default True.
-
struct_mask_axis(Literal['horizontal', 'vertical', 'none'], default:'none') –StructN2V mask axis, by default "none".
-
struct_mask_span(int, default:5) –StructN2V mask span, by default 5.
-
seed(Optional[int], default:None) –Random seed, by default None.