mirrors/mmsegmentation
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmsegmentation.git synced 2025-06-03 22:03:48 +08:00
Code Issues Projects Releases Wiki Activity

Improve structure and readability for FCNHead (#2142)

Browse Source
This commit is contained in:
Edward 2022-10-08 08:31:17 +00:00 committed by GitHub
parent 0391dcd105
commit 2eb13c6906
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
mmseg/models/decode_heads/fcn_head.py
View File

@ -37,28 +37,21 @@ class FCNHead(BaseDecodeHead):
conv_padding = (kernel_size // 2) * dilation conv_padding = (kernel_size // 2) * dilation
convs = [] convs = []
convs.append( for i in range(num_convs):
ConvModule( _in_channels = self.in_channels if i == 0 else self.channels
self.in_channels,
self.channels,
kernel_size=kernel_size,
padding=conv_padding,
dilation=dilation,
conv_cfg=self.conv_cfg,
norm_cfg=self.norm_cfg,
act_cfg=self.act_cfg))
for i in range(num_convs - 1):
convs.append( convs.append(
ConvModule( ConvModule(
self.channels, _in_channels,
self.channels, self.channels,
kernel_size=kernel_size, kernel_size=kernel_size,
padding=conv_padding, padding=conv_padding,
dilation=dilation, dilation=dilation,
conv_cfg=self.conv_cfg, conv_cfg=self.conv_cfg,
norm_cfg=self.norm_cfg, norm_cfg=self.norm_cfg,
act_cfg=self.act_cfg)) act_cfg=self.act_cfg
if num_convs == 0: ))
if len(convs) == 0:
self.convs = nn.Identity() self.convs = nn.Identity()
else: else:
self.convs = nn.Sequential(*convs) self.convs = nn.Sequential(*convs)