Update config.py

添加ocr模块相关配置

pdseg/utils/config.py

@@ -248,7 +248,10 @@ cfg.MODEL.HRNET.STAGE3.NUM_CHANNELS = [40, 80, 160]
 # HRNET STAGE4 设置
 cfg.MODEL.HRNET.STAGE4.NUM_MODULES = 3
 cfg.MODEL.HRNET.STAGE4.NUM_CHANNELS = [40, 80, 160, 320]
+########################## OCNET模型配置 ######################################
 
+cfg.MODEL.OCR.OCR_MID_CHANNELS = 512
+cfg.MODEL.OCR.OCR_KEY_CHANNELS = 256
 ########################## 预测部署模型配置 ###################################
 # 预测保存的模型名称
 cfg.FREEZE.MODEL_FILENAME = '__model__'