tfmri.layers

Keras layers.

Classes

AveragePooling1D

Average pooling for temporal data.

AveragePooling2D

Average pooling operation for spatial data.

AveragePooling3D

Average pooling operation for 3D data (spatial or spatio-temporal).

Conv1D

1D convolution layer (e.g.

Conv2D

2D convolution layer (e.g.

Conv3D

3D convolution layer (e.g.

ConvBlock

A basic convolution block.

DWT1D

Single-level 1D discrete wavelet transform (DWT) layer.

DWT2D

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

DWT3D

Single-level 3D discrete wavelet transform (DWT) layer.

IDWT1D

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

IDWT2D

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

IDWT3D

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

MaxPooling1D

Max pooling operation for 1D temporal data.

MaxPooling2D

Max pooling operation for 2D spatial data.

MaxPooling3D

Max pooling operation for 3D data (spatial or spatio-temporal).

UNet

A UNet layer.

Functions