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) |
Description
GLiftingKernelSE2 creates and uses mask independently from the
maskparam.gconv/gconv/gnn/kernels/kernel_sen/kernel_se2.py
Lines 19 to 45 in 386f30b