gaussian_kernel#
- omnipose.utils.gaussian_kernel(size: int, sigma: float, device=device(type='cuda'))[source]#
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