Update rec_img_aug.py

eshoyuan · Aug 30, 2022 · 1005aaa · 1005aaa
1 parent 3a3fca3
commit 1005aaa
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/ppocr/data/imaug/rec_img_aug.py b/ppocr/data/imaug/rec_img_aug.py
@@ -141,11 +141,7 @@ def __init__(self,
 
     def __call__(self, data):
         img = data['image']
-        if self.infer_mode and self.character_dict_path is not None:
-            norm_img, valid_ratio = resize_norm_img_chinese(img,
-                                                            self.image_shape)
-        else:
-            norm_img, valid_ratio = resize_norm_img(img, self.image_shape,
+        norm_img, valid_ratio = resize_norm_img(img, self.image_shape,
                                                     self.padding)
         data['image'] = norm_img
         data['valid_ratio'] = valid_ratio