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

t changed to expand_ratio.

Browse Source
This commit is contained in:
johnzja 2020-08-12 11:33:09 +08:00
parent b08e1d4e9e
commit 80fd7f6998
1 changed files with 10 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
mmseg/models/backbones/fast_scnn.py
View File

@ -100,7 +100,7 @@ class GlobalFeatureExtractor(nn.Module):
in_channels=64, in_channels=64,
block_channels=(64, 96, 128), block_channels=(64, 96, 128),
out_channels=128, out_channels=128,
t=6, expand_ratio=6,
num_blocks=(3, 3, 3), num_blocks=(3, 3, 3),
pool_scales=(1, 2, 3, 6), pool_scales=(1, 2, 3, 6),
conv_cfg=None, conv_cfg=None,
@ -113,13 +113,13 @@ class GlobalFeatureExtractor(nn.Module):
self.act_cfg = act_cfg self.act_cfg = act_cfg
assert len(block_channels) == len(num_blocks) == 3 assert len(block_channels) == len(num_blocks) == 3
self.bottleneck1 = self._make_layer(in_channels, block_channels[0], self.bottleneck1 = self._make_layer(in_channels, block_channels[0],
num_blocks[0], t, 2) num_blocks[0], expand_ratio, 2)
self.bottleneck2 = self._make_layer(block_channels[0], self.bottleneck2 = self._make_layer(block_channels[0],
block_channels[1], num_blocks[1], block_channels[1], num_blocks[1],
t, 2) expand_ratio, 2)
self.bottleneck3 = self._make_layer(block_channels[1], self.bottleneck3 = self._make_layer(block_channels[1],
block_channels[2], num_blocks[2], block_channels[2], num_blocks[2],
t, 1) expand_ratio, 1)
self.ppm = PPM( self.ppm = PPM(
pool_scales, pool_scales,
block_channels[2], block_channels[2],
@ -136,14 +136,15 @@ class GlobalFeatureExtractor(nn.Module):
norm_cfg=self.norm_cfg, norm_cfg=self.norm_cfg,
act_cfg=self.act_cfg) act_cfg=self.act_cfg)
def _make_layer(self, inplanes, planes, blocks, t=6, stride=1): def _make_layer(self, inplanes, planes, blocks, expand_ratio=6, stride=1):
layers = [] layers = [
layers.append(
InvertedResidual( InvertedResidual(
inplanes, planes, stride, t, norm_cfg=self.norm_cfg)) inplanes, planes, stride, expand_ratio, norm_cfg=self.norm_cfg)
]
for i in range(1, blocks): for i in range(1, blocks):
layers.append( layers.append(
InvertedResidual(planes, planes, 1, t, norm_cfg=self.norm_cfg)) InvertedResidual(
planes, planes, 1, expand_ratio, norm_cfg=self.norm_cfg))
return nn.Sequential(*layers) return nn.Sequential(*layers)
def forward(self, x): def forward(self, x):