From 08c59b0fdc9fef874ae134e500f1dadd8c3b9995 Mon Sep 17 00:00:00 2001
From: tianlanshidai <48075497+tianlanshidai@users.noreply.github.com>
Date: Wed, 15 Jul 2020 18:30:03 +0800
Subject: [PATCH] Update config.py
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

添加ocr模块相关配置
---
 pdseg/utils/config.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/pdseg/utils/config.py b/pdseg/utils/config.py
index 8c1bccc1..39540fcb 100644
--- a/pdseg/utils/config.py
+++ b/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__'
-- 
GitLab