pado.math.gaussian_window

gaussian_window(size, sigma)[source]

Create normalized 2D Gaussian window.

Parameters:
  • size (int) – Width and height of square window

  • sigma (float) – Standard deviation of Gaussian

Returns:

Normalized 2D Gaussian window [size, size]

Return type:

torch.Tensor

Examples

>>> window = gaussian_window(11, 1.5)