Fix bug

pdseg/models/modeling/deeplab.py

@@ -274,12 +274,14 @@ def _mobilenetv3(input, model_name='large'):
     # OUTPUT_STRIDE：下采样倍数
     scale = cfg.MODEL.DEEPLAB.DEPTH_MULTIPLIER
     output_stride = cfg.MODEL.DEEPLAB.OUTPUT_STRIDE
-    lr_mult_shortcut = cfg.MODEL.DEEPLAB.BACKBONE_LR_MULT_LIST
+    lr_mult_list = cfg.MODEL.DEEPLAB.BACKBONE_LR_MULT_LIST
+    if lr_mult_list is None:
+        lr_mult_list = [1.0, 1.0, 1.0, 1.0, 1.0]
     model = mobilenet_v3_backbone(
         scale=scale,
         output_stride=output_stride,
         model_name=model_name,
-        lr_mult_list=lr_mult_shortcut)
+        lr_mult_list=lr_mult_list)
     data, decode_shortcut = model.net(input)
     return data, decode_shortcut
 

tutorial/finetune_ocrnet.md

@@ -77,7 +77,9 @@ MODEL:
         STAGE4:
             NUM_CHANNELS: [18, 36, 72, 144]
     # 此处设置OCR HEAD
-    
+        OCR_MID_CHANNELS: 512
+        OCR_KEY_CHANNELS: 256
+    MULTI_LOSS_WEIGHT: [1.0, 1.0]
 
 # 其他配置
 TRAIN_CROP_SIZE: (512, 512)
@@ -142,4 +144,4 @@ python pdseg/vis.py --use_gpu --cfg ./configs/ocrnet_optic.yaml
 
 |预训练模型名称|Backbone|数据集|配置|
 |-|-|-|-|
-|ocrnet_w18_bn_cityscapes|OCRNet| Cityscapes | MODEL.MODEL_NAME: ocrnet <br> MODEL.HRNET.STAGE2.NUM_CHANNELS: [18, 36] <br> MODEL.HRNET.STAGE3.NUM_CHANNELS: [18, 36, 72] <br> MODEL.HRNET.STAGE4.NUM_CHANNELS: [18, 36, 72, 144] <br> MODEL.DEFAULT_NORM_TYPE: bn <br> MODEL.OCR.OCR_MID_CHANNELS: 512 <br> MODEL.OCR.OCR_KEY_CHANNELS: 256|
+|ocrnet_w18_bn_cityscapes|OCRNet| Cityscapes | MODEL.MODEL_NAME: ocrnet <br> MODEL.HRNET.STAGE2.NUM_CHANNELS: [18, 36] <br> MODEL.HRNET.STAGE3.NUM_CHANNELS: [18, 36, 72] <br> MODEL.HRNET.STAGE4.NUM_CHANNELS: [18, 36, 72, 144] <br> MODEL.DEFAULT_NORM_TYPE: bn <br> MODEL.OCR.OCR_MID_CHANNELS: 512 <br> MODEL.OCR.OCR_KEY_CHANNELS: 256 <br> MODEL.MULTI_LOSS_WEIGHT: [1.0, 1.0] |