fix ci export bug

configs/rec/rec_r45_abinet.yml

@@ -72,7 +72,6 @@ Train:
           ignore_index: *ignore_index
       - ABINetRecResizeImg:
           image_shape: [3, 32, 128]
-          padding: False
       - KeepKeys:
           keep_keys: ['image', 'label', 'length'] # dataloader will return list in this order
   loader:
@@ -93,7 +92,6 @@ Eval:
           ignore_index: *ignore_index
       - ABINetRecResizeImg:
           image_shape: [3, 32, 128]
-          padding: False
       - KeepKeys:
           keep_keys: ['image', 'label', 'length'] # dataloader will return list in this order
   loader:

configs/rec/rec_svtrnet.yml

@@ -84,7 +84,6 @@ Train:
           channel_first: False
       - CTCLabelEncode: # Class handling label
       - SVTRRecResizeImg:
-          character_dict_path:
           image_shape: [3, 64, 256]
           padding: False
       - KeepKeys:
@@ -105,7 +104,6 @@ Eval:
           channel_first: False
       - CTCLabelEncode: # Class handling label
       - SVTRRecResizeImg:
-          character_dict_path:
           image_shape: [3, 64, 256]
           padding: False
       - KeepKeys:

configs/rec/rec_svtrnet_ch.yml

@@ -110,8 +110,6 @@ Train:
     - RecAug: null
     - CTCLabelEncode: null
     - SVTRRecResizeImg:
-        character_dict_path: null
-        infer_mode: False
         image_shape:
         - 3
         - 32
@@ -139,8 +137,6 @@ Eval:
         channel_first: false
     - CTCLabelEncode: null
     - SVTRRecResizeImg:
-        character_dict_path: null
-        infer_mode: False
         image_shape:
         - 3
         - 32

ppocr/data/imaug/rec_img_aug.py

@@ -312,14 +312,8 @@ class GrayRecResizeImg(object):
 
 
 class ABINetRecResizeImg(object):
-    def __init__(self,
+    def __init__(self, image_shape, **kwargs):
-                 image_shape,
-                 infer_mode=False,
-                 character_dict_path=None,
-                 **kwargs):
         self.image_shape = image_shape
-        self.infer_mode = infer_mode
-        self.character_dict_path = character_dict_path
 
     def __call__(self, data):
         img = data['image']
@@ -330,15 +324,8 @@ class ABINetRecResizeImg(object):
 
 
 class SVTRRecResizeImg(object):
-    def __init__(self,
+    def __init__(self, image_shape, padding=True, **kwargs):
-                 image_shape,
-                 infer_mode=False,
-                 character_dict_path='./ppocr/utils/ppocr_keys_v1.txt',
-                 padding=True,
-                 **kwargs):
         self.image_shape = image_shape
-        self.infer_mode = infer_mode
-        self.character_dict_path = character_dict_path
         self.padding = padding
 
     def __call__(self, data):

test_tipc/configs/rec_svtrnet/rec_svtrnet.yml

@@ -84,8 +84,7 @@ Train:
           img_mode: BGR
           channel_first: False
       - CTCLabelEncode: # Class handling label
-      - RecResizeImg:
+      - SVTRRecResizeImg:
-          character_dict_path:
           image_shape: [3, 64, 256]
           padding: False
       - KeepKeys:
@@ -106,8 +105,7 @@ Eval:
           img_mode: BGR
           channel_first: False
       - CTCLabelEncode: # Class handling label
-      - RecResizeImg:
+      - SVTRRecResizeImg:
-          character_dict_path:
           image_shape: [3, 64, 256]
           padding: False
       - KeepKeys: