提交 ed400c86 编写于 作者: W WenmuZhou

mv3 small support det

上级 0275354c
......@@ -32,6 +32,7 @@ class MobileNetV3():
"""
self.scale = params['scale']
model_name = params['model_name']
self.model_name = model_name
self.inplanes = 16
if model_name == "large":
self.cfg = [
......@@ -80,7 +81,7 @@ class MobileNetV3():
"supported scale are {} but input scale is {}".format(supported_scale, self.scale)
self.disable_se = params.get('disable_se', False)
def __call__(self, input):
scale = self.scale
inplanes = self.inplanes
......@@ -102,7 +103,8 @@ class MobileNetV3():
inplanes = self.make_divisible(inplanes * scale)
outs = []
for layer_cfg in cfg:
if layer_cfg[5] == 2 and i > 2:
start_idx = 2 if self.model_name == 'large' else 0
if layer_cfg[5] == 2 and i > start_idx:
outs.append(conv)
conv = self.residual_unit(
input=conv,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册