未验证 提交 e77baea4 编写于 作者: K Kaipeng Deng 提交者: GitHub

check extra_conv needed in mobilenet_v3 (#933)

上级 26c2a4f2
...@@ -393,20 +393,22 @@ class MobileNetV3(object): ...@@ -393,20 +393,22 @@ class MobileNetV3(object):
self.end_points.append(conv) self.end_points.append(conv)
# extra block # extra block
conv_extra = self._conv_bn_layer( # check whether conv_extra is needed
conv, if self.block_stride < max(self.feature_maps):
filter_size=1, conv_extra = self._conv_bn_layer(
num_filters=self._make_divisible(scale * cfg[-1][1]), conv,
stride=1, filter_size=1,
padding="SAME", num_filters=self._make_divisible(scale * cfg[-1][1]),
num_groups=1, stride=1,
if_act=True, padding="SAME",
act='hard_swish', num_groups=1,
name='conv' + str(i + 2)) if_act=True,
self.block_stride += 1 act='hard_swish',
if self.block_stride in self.feature_maps: name='conv' + str(i + 2))
self.end_points.append(conv_extra) self.block_stride += 1
i += 1 if self.block_stride in self.feature_maps:
self.end_points.append(conv_extra)
i += 1
for block_filter in self.extra_block_filters: for block_filter in self.extra_block_filters:
conv_extra = self._extra_block_dw(conv_extra, block_filter[0], conv_extra = self._extra_block_dw(conv_extra, block_filter[0],
block_filter[1], 2, block_filter[1], 2,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册