diff --git a/ppocr/modeling/backbones/det_mobilenet_v3.py b/ppocr/modeling/backbones/det_mobilenet_v3.py index 508f2bbf2d73216fc35a2d906f94731707a25736..0b1f5dc0cb107d9f6f646c8449d4886d4f63c4f6 100755 --- a/ppocr/modeling/backbones/det_mobilenet_v3.py +++ b/ppocr/modeling/backbones/det_mobilenet_v3.py @@ -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,