diff --git a/mmseg/models/decode_heads/fcn_head.py b/mmseg/models/decode_heads/fcn_head.py index 0ba09a010..e27be6923 100644 --- a/mmseg/models/decode_heads/fcn_head.py +++ b/mmseg/models/decode_heads/fcn_head.py @@ -48,9 +48,8 @@ class FCNHead(BaseDecodeHead): dilation=dilation, conv_cfg=self.conv_cfg, norm_cfg=self.norm_cfg, - act_cfg=self.act_cfg - )) - + act_cfg=self.act_cfg)) + if len(convs) == 0: self.convs = nn.Identity() else: