`unet`¶

Original U-Net model implementation.

Module Contents¶

class unet.Model(hparams)¶

Bases: model.base_model.BaseModel

Inheritance diagram of unet.Model

This class implements U-Net module and its training functionality. It is equivalent: to PyTorch’s nn.Module in all aspects.

Parameters: LightningModule (nn.Module) – The Pytorch-Lightning module derived from nn.module with useful hooks

forward(self, x)¶

Does the forward pass on the model.

Parameters: x (torch.Tensor) – Input tensor batch.
Returns: Output activations.
Return type: torch.Tensor