From e224dc10eb62b4d771aafc1ffcb787eedb35ba7b Mon Sep 17 00:00:00 2001
From: Kai Chen <chenkaidev@gmail.com>
Date: Thu, 22 Aug 2019 17:15:26 +0800
Subject: [PATCH] add an argument no_norm_on_lateral to fpn (#1240)
---
mmdet/models/necks/fpn.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/mmdet/models/necks/fpn.py b/mmdet/models/necks/fpn.py
index d42fb1d..77dd409 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(
--
GitLab