Update paddleocr.py (#10290)

解决多次识别pdf，受第一次页面最大数量影响的bug
2023-07-04 15:42:07 +08:00 · 2023-07-04 15:42:07 +08:00 · b8888627a5
parent 1fa3575030
commit b8888627a5
1 changed files with 3 additions and 2 deletions
--- a/paddleocr.py
+++ b/paddleocr.py
@ -532,8 +532,9 @@ class PaddleOCR(predict_system.TextSystem):
        # for infer pdf file
        if isinstance(img, list):
            if self.page_num > len(img) or self.page_num == 0:
-                self.page_num = len(img)
-            imgs = img[:self.page_num]
+                imgs=img
+            else:
+                imgs = img[:self.page_num]
        else:
            imgs = [img]
        if det and rec: