Function get_window

utils/general_util.py:271–289

Return a window function. Args: win_type (str): Window type. Can either be one of the window function provided in PyTorch ['hann_window', 'bartlett_window', 'blackman_window', 'hamming_window', 'kaiser_window'] or any of the windows provided by [SciPy](https://do

(win_type: str, win_length: int)

    return count

def get_window(win_type: str, win_length: int):
    """Return a window function.

    Args:
        win_type (str): Window type. Can either be one of the window function provided in PyTorch
            ['hann_window', 'bartlett_window', 'blackman_window', 'hamming_window', 'kaiser_window']
            or any of the windows provided by [SciPy](https://docs.scipy.org/doc/scipy/reference/generated/scipy.signal.windows.get_window.html).
        win_length (int): Window length

    Returns:
        win: The window as a 1D torch tensor
    """

    try:
        win = getattr(torch, win_type)(win_length)
    except:
        win = torch.from_numpy(scipy.signal.windows.get_window(win_type, win_length))

    return win


# https://github.com/facebookresearch/pytorch3d/blob/main/pytorch3d/transforms/rotation_conversions.py