upodate deeplab.py

pdseg/models/backbone/resnet_vd.py

@@ -18,6 +18,7 @@ from __future__ import division
 from __future__ import print_function
 
 import math
+
 import numpy as np
 import paddle.fluid as fluid
 from paddle.fluid.param_attr import ParamAttr

pdseg/models/modeling/deeplab.py

@@ -265,8 +265,11 @@ def deeplabv3p(img, num_classes):
     # Backbone设置：xception 或 mobilenetv2
     if 'xception' in cfg.MODEL.DEEPLAB.BACKBONE:
         data, decode_shortcut = xception(img)
+        print('xception backbone do not support BACKBONE_LR_MULT_LIST setting')
     elif 'mobilenet' in cfg.MODEL.DEEPLAB.BACKBONE:
         data, decode_shortcut = mobilenetv2(img)
+        print(
+            'mobilenetv2 backbone do not support BACKBONE_LR_MULT_LIST setting')
     elif 'resnet' in cfg.MODEL.DEEPLAB.BACKBONE:
         data, decode_shortcut = resnet_vd(img)
     else:

pdseg/utils/config.py

@@ -209,7 +209,7 @@ cfg.MODEL.DEEPLAB.ASPP_WITH_SEP_CONV = True
 # 解码器是否使用可分离卷积
 cfg.MODEL.DEEPLAB.DECODER_USE_SEP_CONV = True
 # resnet_vd分阶段学习率
-cfg.MODEL.DEEPLAB.RESNET_LR_MULT_LIST = [1.0, 1.0, 1.0, 1.0, 1.0]
+cfg.MODEL.DEEPLAB.BACKBONE_LR_MULT_LIST = [1.0, 1.0, 1.0, 1.0, 1.0]
 
 ########################## UNET模型配置 #######################################
 # 上采样方式, 默认为双线性插值