Pfhead (#9898)

* support min_area_rect crop * add check_install * fix requirement.txt * fix check_install * add lanms-neo for drrg * fix * fix doc * fix * support set gpu_id when inference * fix #8855 * fix #8855 * opt slim doc * fix doc bug * rename * rename
PaddlePaddle · May 15, 2023 · 24ff4de · 24ff4de
1 parent dd8cc21
commit 24ff4de
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/configs/det/ch_PP-OCRv4/ch_PP-OCRv4_det_student.yml b/configs/det/ch_PP-OCRv4/ch_PP-OCRv4_det_student.yml
@@ -31,7 +31,7 @@ Architecture:
     out_channels: 96
     shortcut: True
   Head:
-    name: CBNHeadLocal
+    name: PFHeadLocal
     k: 50
     mode: "small"
 

diff --git a/configs/det/ch_PP-OCRv4/ch_PP-OCRv4_det_teacher.yml b/configs/det/ch_PP-OCRv4/ch_PP-OCRv4_det_teacher.yml
@@ -30,7 +30,7 @@ Architecture:
     out_channels: 256
     intracl: true
   Head:
-    name: CBNHeadLocal
+    name: PFHeadLocal
     k: 50
     mode: "large"
 

diff --git a/ppocr/modeling/heads/__init__.py b/ppocr/modeling/heads/__init__.py
@@ -17,7 +17,7 @@
 
 def build_head(config):
     # det head
-    from .det_db_head import DBHead, CBNHeadLocal
+    from .det_db_head import DBHead, PFHeadLocal
     from .det_east_head import EASTHead
     from .det_sast_head import SASTHead
     from .det_pse_head import PSEHead
@@ -56,7 +56,7 @@ def build_head(config):
         'TableAttentionHead', 'SARHead', 'AsterHead', 'SDMGRHead', 'PRENHead',
         'MultiHead', 'ABINetHead', 'TableMasterHead', 'SPINAttentionHead',
         'VLHead', 'SLAHead', 'RobustScannerHead', 'CT_Head', 'RFLHead',
-        'DRRGHead', 'CANHead', 'SATRNHead', 'CBNHeadLocal'
+        'DRRGHead', 'CANHead', 'SATRNHead', 'PFHeadLocal'
     ]
 
     if config['name'] == 'DRRGHead':

diff --git a/ppocr/modeling/heads/det_db_head.py b/ppocr/modeling/heads/det_db_head.py
@@ -129,9 +129,9 @@ def forward(self, x, init_map, distance_map):
         return out
 
 
-class CBNHeadLocal(DBHead):
+class PFHeadLocal(DBHead):
     def __init__(self, in_channels, k=50, mode='small', **kwargs):
-        super(CBNHeadLocal, self).__init__(in_channels, k, **kwargs)
+        super(PFHeadLocal, self).__init__(in_channels, k, **kwargs)
         self.mode = mode
 
         self.up_conv = nn.Upsample(scale_factor=2, mode="nearest", align_mode=1)