def __init__()

in utils/gluon/utils/resnetv2.py [0:0]

• 37 lines of code
• 11 McCabe index (conditional complexity)

    def __init__(self, in_planes, mid_planes, out_planes, groups=1, strides=1,
                 norm_kwargs=None, name_prefix=None,
                 down_pos=0, use_se=False, se_planes=-1, **kwargs):
        super(_BottleneckV2, self).__init__(prefix=name_prefix)
        assert down_pos in [0, 1, 2], \
            "down_pos value({}) is unknown.".format(down_pos)
        strides1 = strides if down_pos == 0 else 1
        strides2 = strides if down_pos == 1 else 1
        strides3 = strides if down_pos == 2 else 1
        with self.name_scope():
            # extract information
            self.bn1 = nn.BatchNorm(in_channels=in_planes, prefix='bn1',
                                  **({} if norm_kwargs is None else norm_kwargs))
            self.relu1 = nn.Activation('relu')
            self.conv1 = nn.Conv2D(channels=mid_planes, in_channels=in_planes,
                                  kernel_size=1, use_bias=False, strides=strides1,
                                  prefix='conv1')
            # capture spatial relations
            self.bn2 = nn.BatchNorm(in_channels=mid_planes, prefix='bn2',
                                  **({} if norm_kwargs is None else norm_kwargs))
            self.relu2 = nn.Activation('relu')
            self.conv2 = nn.Conv2D(channels=mid_planes, in_channels=mid_planes,
                                  kernel_size=3, padding=1, groups=groups,
                                  strides=strides2, use_bias=False, prefix='conv2')
            # embeding back to information highway
            self.bn3 = nn.BatchNorm(in_channels=mid_planes, prefix='bn3',
                                  **({} if norm_kwargs is None else norm_kwargs))
            self.relu3 = nn.Activation('relu')
            self.conv3 = nn.Conv2D(channels=out_planes, in_channels=mid_planes,
                                  kernel_size=1, use_bias=False, strides=strides3,
                                  prefix='conv3')

            self.se_block = nn.SE(in_channels=out_planes, channels=se_planes,
                                  prefix='se') if use_se else None

            if strides != 1 or in_planes != out_planes:
                self.bn4 = nn.BatchNorm(in_channels=in_planes, prefix='bn4',
                                  **({} if norm_kwargs is None else norm_kwargs))
                self.relu4 = nn.Activation('relu')
                self.conv4 = nn.Conv2D(channels=out_planes, in_channels=in_planes,
                                  kernel_size=1, strides=strides,
                                  prefix='conv4')