Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

GLiftingKernelSE2 ignores mask=False #3

Open
dgcnz opened this issue May 13, 2024 · 0 comments
Open

GLiftingKernelSE2 ignores mask=False #3

dgcnz opened this issue May 13, 2024 · 0 comments

Comments

@dgcnz
Copy link

dgcnz commented May 13, 2024

Description

GLiftingKernelSE2 creates and uses mask independently from the mask param.

mask: bool = True,
grid_H: Tensor | None = None,
) -> None:
"""
Implements SE2 lifting kernel.
Arguments:
- in_channels: int denoting the number of input channels.
- out_channels: int denoting the number of output channels.
- kernel_size: int denoting the spatial kernel size.
- group_kernel_size: int denoting the group kernel size.
- groups: number of groups for depth-wise separability.
- sampling_mode: str indicating the sampling mode. Supports bilinear (default)
or nearest.
- sampling_padding_mode: str indicating padding mode for sampling. Default
border.
- mask: bool if true, will initialize spherical mask.
- grid_H: tensor of reference grid used for interpolation. If not
provided, a uniform grid of group_kernel_size will be
generated. If provided, will overwrite given group_kernel_size.
"""
if grid_H is None:
grid_H = so2.uniform_grid(group_kernel_size)
grid_Rn = gF.create_grid_R2(kernel_size)
mask = gF.create_spherical_mask_R2(kernel_size)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant