locpix.img_processing.models.unet#
UNET model
Adapted from milesial/Pytorch-UNet
Classes
|
(convolution => [BN] => ReLU) * 2 |
|
Downscaling with maxpool then double conv |
|
Initialize internal Module state, shared by both nn.Module and ScriptModule. |
|
Upscaling then double conv |
|
Initialize internal Module state, shared by both nn.Module and ScriptModule. |