group1_arch_branch

tiatoolbox . models . architecture . micronet . group1_arch_branch

group1_arch_branch(in_ch, resized_in_ch, out_ch)

Group1 branch for MicroNet.

Parameters

• in_ch (int) – Number of input channels.

• resized_in_ch (int) – Number of input channels from resized input.

• out_ch (int) – Number of output channels.

Returns

An output of type torch.nn.ModuleDict

Return type

torch.nn.ModuleDict