cellmap_data.transforms.augment.gaussian_blur

Classes

GaussianBlur([kernel_size, sigma, dim])

Initialize a Gaussian Blur module.

class cellmap_data.transforms.augment.gaussian_blur.GaussianBlur(kernel_size: int = 3, sigma: float = 0.1, dim: int = 2)

Initialize a Gaussian Blur module.

Parameters:

• kernel_size (int) – Size of the Gaussian kernel (should be odd).

• sigma (float) – Standard deviation of the Gaussian distribution.

• dim (int) – Dimensionality (2 or 3) for applying the blur.

forward(x: Tensor)

Apply Gaussian blur to the input tensor.

Parameters:

x (Tensor)