fusionlab.segmentation.tfbase module#

class fusionlab.segmentation.tfbase.TFSegmentationModel(*args, **kwargs)[source]#

Bases: Model

Base PyTorch class of the segmentation model with Encoder, Bridger, Decoder, Head

call(x, training=None)[source]#
Parameters:

  • x – input tensor

  • training – flag for BatchNormalization and Dropout, whether the layer should behave in training mode or in

  • mode (inference) –

Returns: