gaussian_kernel

omnipose.utils.gaussian_kernel(size: int, sigma: float)

Creates a 2D Gaussian kernel with mean 0.

Parameters

• size (int) -- The size of the kernel. Should be an odd number.

• sigma (float) -- The standard deviation of the Gaussian distribution.

Returns

The Gaussian kernel.

Return type

torch.Tensor