Update mmseg/models/decode_heads/sep_fcn_head.py

Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com>
2020-08-12 15:30:39 +08:00 · 2020-08-12 15:30:39 +08:00 · 568106c2e2
parent 95d1a8d8d7
commit 568106c2e2
1 changed files with 2 additions and 2 deletions
--- a/mmseg/models/decode_heads/sep_fcn_head.py
+++ b/mmseg/models/decode_heads/sep_fcn_head.py
@ -30,8 +30,8 @@ class DepthwiseSeparableFCNHead(FCNHead):
        self.convs[0] = DepthwiseSeparableConvModule(
            self.in_channels,
            self.channels,
-            kernel_size=3,
-            padding=1,
+            kernel_size=self.kernel_size,
+            padding=self.kernel_size//2,
            norm_cfg=self.norm_cfg)
        for i in range(1, self.num_convs):
            self.convs[i] = DepthwiseSeparableConvModule(