Skip to content

N2V Manipulate

Source

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.