group4_arch_branch

tiatoolbox . models . architecture . micronet . group4_arch_branch

group4_arch_branch(in_ch, out_ch, up_kernel=(2, 2), up_strides=(2, 2), activation='tanh')

Group4 branch for MicroNet.

This branch defines architecture for decoder and provides input for the auxiliary and main output branch.

Parameters:

• in_ch (int) – Number of input channels.

• out_ch (int) – Number of output channels.

• up_kernel (tuple of int) – Kernel size for torch.nn.ConvTranspose2d.

• up_strides (tuple of int) – Stride size for torch.nn.ConvTranspose2d.

• activation (str) – Activation function, default=”tanh”.

Returns:

An output of type torch.nn.ModuleDict

Return type:

torch.nn.ModuleDict