提交 84b0e6e8 编写于 作者: J jiangjiajun

fix bug for yolov3 mobilenetv3

上级 5f3974c7
......@@ -95,9 +95,9 @@ class YOLOv3(BaseAPI):
elif backbone_name == 'MobileNetV1':
backbone = paddlex.cv.nets.MobileNetV1(norm_type='sync_bn')
elif backbone_name.startswith('MobileNetV3'):
models_name = backbone_name.split('_')[1]
model_name = backbone_name.split('_')[1]
backbone = paddlex.cv.nets.MobileNetV3(
norm_type='sync_bn', models_name=models_name)
norm_type='sync_bn', model_name=model_name)
return backbone
def build_net(self, mode='train'):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册