Method __init__

PATH/core/models/necks/simple_fpn.py:535–569

(self,
                 num_feature_levels,
                 hidden_dim,
                 backbone, # placeholder
                 bn_group,
                 **kwargs)

class PedDetSimpleResNetFPN(nn.Module):
    def __init__(self,
                 num_feature_levels,
                 hidden_dim,
                 backbone, # placeholder
                 bn_group,
                 **kwargs):
        super(PedDetSimpleResNetFPN, self).__init__()
        num_backbone_outs = len(backbone.strides)
        self.backbone = [backbone]
        self.num_feature_levels = num_feature_levels
        if num_feature_levels > 1:
            num_backbone_outs = len(backbone.strides)
            input_proj_list = []
            for _ in range(num_backbone_outs):
                in_channels = backbone.num_channels[_]
                input_proj_list.append(nn.Sequential(
                    nn.Conv2d(in_channels, hidden_dim, kernel_size=1),
                    nn.GroupNorm(32, hidden_dim),
                ))
            for _ in range(num_feature_levels - num_backbone_outs):
                input_proj_list.append(nn.Sequential(
                    nn.Conv2d(in_channels, hidden_dim, kernel_size=3, stride=2, padding=1),
                    nn.GroupNorm(32, hidden_dim),
                ))
                in_channels = hidden_dim
            self.input_proj = nn.ModuleList(input_proj_list)
        else:
            self.input_proj = nn.ModuleList([
                nn.Sequential(
                    nn.Conv2d(backbone.num_channels[-1], hidden_dim, kernel_size=1),
                    nn.GroupNorm(32, hidden_dim),
                )
            ])

        self._reset_parameters()

    def _reset_parameters(self):
        for proj in self.input_proj: