fix(table): improve table image processing

- Replace np.array with np.asarray for better performance - Add image color conversion from RGB to BGR using OpenCV

fix(table): improve table image processing
- Replace np.array with np.asarray for better performance - Add image color conversion from RGB to BGR using OpenCV
401dfa4e · myhloli · 376907ef · 401dfa4e
Commit 401dfa4e authored Nov 05, 2024 by myhloli
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

ppTableModel.py magic_pdf/model/ppTableModel.py +3 -1

No files found.
--- a/magic_pdf/model/ppTableModel.py
+++ b/magic_pdf/model/ppTableModel.py
+import cv2
 from paddleocr.ppstructure.table.predict_table import TableSystem
 from paddleocr.ppstructure.utility import init_args
 from magic_pdf.libs.Constants import *
@@ -36,7 +37,8 @@ class ppTableModel(object):
        - HTML (str): A string representing the HTML structure with content of the table.
        """
        if isinstance(image, Image.Image):
-            image = np.array(image)
+            image = np.asarray(image)
+            image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
        pred_res, _ = self.table_sys(image)
        pred_html = pred_res["html"]
        # res = '<td><table  border="1">' + pred_html.replace("<html><body><table>", "").replace(