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))

Group4 branch for MicroNet.

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.

Returns

An output of type torch.nn.ModuleDict

Return type

torch.nn.ModuleDict