diff --git a/mmdet/models/necks/fpn.py b/mmdet/models/necks/fpn.py index d42fb1d59f701278fb8d7b0ad24c7ef76db184ba..77dd409c437a5f12852090e101c71eaadedfc8ed 100644 --- a/mmdet/models/necks/fpn.py +++ b/mmdet/models/necks/fpn.py @@ -19,6 +19,7 @@ class FPN(nn.Module): add_extra_convs=False, extra_convs_on_inputs=True, relu_before_extra_convs=False, + no_norm_on_lateral=False, conv_cfg=None, norm_cfg=None, activation=None): @@ -30,6 +31,7 @@ class FPN(nn.Module): self.num_outs = num_outs self.activation = activation self.relu_before_extra_convs = relu_before_extra_convs + self.no_norm_on_lateral = no_norm_on_lateral self.fp16_enabled = False if end_level == -1: @@ -54,7 +56,7 @@ class FPN(nn.Module): out_channels, 1, conv_cfg=conv_cfg, - norm_cfg=norm_cfg, + norm_cfg=norm_cfg if not self.no_norm_on_lateral else None, activation=self.activation, inplace=False) fpn_conv = ConvModule(