spyrit.core.nnet.Unet

class spyrit.core.nnet.Unet(in_channel=1, out_channel=1)

Bases: Module

Methods

bottle_neck(in_channels[, kernel_size, padding])
concat(upsampled, bypass)
contract(in_channels, out_channels[, ...])
expans(in_channels, mid_channel, out_channels)
final_block(in_channels, mid_channel, ...[, ...])
forward(x)	Define the computation performed at every call.