trw.train.filter_gaussian

Module Contents

Classes

FilterFixed

Apply a fixed filter to n-dimensional images

FilterGaussian

Implement a gaussian filter as a torch.nn.Module
class trw.train.filter_gaussian.FilterFixed(kernel: torch.Tensor, groups: int = 1, padding: int = 0)

Bases: torch.nn.Module

Apply a fixed filter to n-dimensional images

__call__(self, value: trw.basic_typing.TorchTensorNCX) → trw.basic_typing.TorchTensorNCX
class trw.train.filter_gaussian.FilterGaussian(input_channels: int, nb_dims: int, sigma: Union[float, Sequence[float]], kernel_sizes: Optional[Union[int, Sequence[int]]] = None, padding: typing_extensions.Literal[same, none] = 'same', device: Optional[torch.device] = None)

Bases: FilterFixed

Implement a gaussian filter as a torch.nn.Module