moonshine.models.UNet

class moonshine.models.UNet(name)

A basic UNet model, implemented under the hood with segmentation-models- pytorch.

The model uses a ResNet backbone and skip connections, as in the original paper at https://arxiv.org/abs/1505.04597. Some parameters are fixed, since pre-trained weights require a consistent network structure.