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, groups=1, padding=0)

Bases: torch.nn.Module

Apply a fixed filter to n-dimensional images

__call__(self, value)

class trw.train.filter_gaussian.FilterGaussian(input_channels, nb_dims, sigma, kernel_sizes=None, padding='same', device=None)

Bases: FilterFixed

Implement a gaussian filter as a torch.nn.Module