tfmri.layers.IDWT2D

class IDWT2D(*args, **kwargs)[source]

Bases: tensorflow_mri.python.layers.signal_layers.DWT

Single-level 2D inverse discrete wavelet transform (IDWT) layer.

The input format is determined by the format_dict argument. If format_dict is True (default), the input should be a dict with keys 'aa', 'ad', 'da', 'dd', where 'a' is for approximation and 'd' is for detail. The value for each key should be a tensor of shape [batch_size, height, width, channels]. If format_dict is False, the input should be a list of tensors corresponding to each of the keys above.

The output is a tensor of shape [batch_size, out_height, out_width, channels].

Parameters

  • wavelet –

    A str or a length-2 list of str. When passed a list, different wavelets are applied along each axis.

  • mode –

    A str. The padding or signal extension mode. Must be one of the values supported by tfmri.signal.dwt. Defaults to 'symmetric'.

  • format_dict –

    A boolean. If True, the input is a dict. Otherwise, it is a list.