create_block#
tiatoolbox
.models
.architecture
.unet
.create_block
- create_block(pre_activation, kernels, input_ch, output_ch)[source]#
Helper to create a block of Vanilla Convolution.
This is in pre-activation style.
- Parameters:
pre_activation (bool) – Whether to apply activation layer before the convolution layer. Should be True for ResNet blocks.
kernels (list) – A list of convolution layers. Each item is an integer and denotes the layer kernel size.
input_ch (int) – Number of channels in the input images.
output_ch (int) – Number of channels in the output images.