Merge pull request #2071 from myhloli/dev

refactor(ocr): remove redundant code and improve code quality

magic_pdf/model/sub_modules/ocr/paddleocr2pytorch/pytorch_paddle.py

 # Copyright (c) Opendatalab. All rights reserved.
 import copy
 import os.path
+import warnings
 from pathlib import Path
 
 import cv2
@@ -92,7 +93,8 @@ class PytorchPaddleOCR(TextSystem):
             exit(0)
         img = check_img(img)
         imgs = [img]
-
+        with warnings.catch_warnings():
+            warnings.simplefilter("ignore", category=RuntimeWarning)
             if det and rec:
                 ocr_res = []
                 for img in imgs:

magic_pdf/model/sub_modules/ocr/paddleocr2pytorch/tools/infer/predict_rec.py

@@ -371,12 +371,6 @@ class TextRecognizer(BaseOCRV20):
                     gsrm_slf_attn_bias1_inp = torch.from_numpy(gsrm_slf_attn_bias1_list)
                     gsrm_slf_attn_bias2_inp = torch.from_numpy(gsrm_slf_attn_bias2_list)
 
-                    # if self.use_gpu:
-                    #     inp = inp.cuda()
-                    #     encoder_word_pos_inp = encoder_word_pos_inp.cuda()
-                    #     gsrm_word_pos_inp = gsrm_word_pos_inp.cuda()
-                    #     gsrm_slf_attn_bias1_inp = gsrm_slf_attn_bias1_inp.cuda()
-                    #     gsrm_slf_attn_bias2_inp = gsrm_slf_attn_bias2_inp.cuda()
                     inp = inp.to(self.device)
                     encoder_word_pos_inp = encoder_word_pos_inp.to(self.device)
                     gsrm_word_pos_inp = gsrm_word_pos_inp.to(self.device)
@@ -398,8 +392,6 @@ class TextRecognizer(BaseOCRV20):
 
                 with torch.no_grad():
                     inp = torch.from_numpy(norm_img_batch)
-                    # if self.use_gpu:
-                    #     inp = inp.cuda()
                     inp = inp.to(self.device)
                     preds = self.net(inp)