unet_layer#

Layer for unet

Classes

ConvbrLayer(*args, **kwargs)

1D Convolution + BatchNorm + ReLU block.

ReBlock(*args, **kwargs)

Residual block with two Convbr layers and optional SE block.

SeBlock(*args, **kwargs)

Squeeze-and-Excitation block for channel-wise attention.