From 273068527f9c7217ff9af34df4a20dbd378b4841 Mon Sep 17 00:00:00 2001
From: WenmuZhou <572459439@qq.com>
Date: Tue, 18 Oct 2022 07:31:07 +0000
Subject: [PATCH] only in DRRG import DRRGHead

---
 ppocr/modeling/heads/__init__.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/ppocr/modeling/heads/__init__.py b/ppocr/modeling/heads/__init__.py
index 63002140..c203af01 100755
--- a/ppocr/modeling/heads/__init__.py
+++ b/ppocr/modeling/heads/__init__.py
@@ -24,7 +24,6 @@ def build_head(config):
     from .det_fce_head import FCEHead
     from .e2e_pg_head import PGHead
     from .det_ct_head import CT_Head
-    from .det_drrg_head import DRRGHead
 
     # rec head
     from .rec_ctc_head import CTCHead
@@ -59,6 +58,10 @@ def build_head(config):
         'DRRGHead'
     ]
 
+    if config['name'] == 'DRRGHead':
+        from .det_drrg_head import DRRGHead
+        support_dict.append('DRRGHead')
+
     #table head
 
     module_name = config.pop('name')
-- 
GitLab