out_arch_branch

tiatoolbox . models . architecture . micronet . out_arch_branch

out_arch_branch(in_ch, num_class=2)

Group5 branch for MicroNet.

Parameters

• in_ch (int) – Number of input channels.

• num_class (int) – Number of output channels. default=2.

Returns

An output of type torch.nn.Sequential

Return type

torch.nn.Sequential