Merge pull request #866 from myhloli/dev

fix(table): improve table image processing

Merge pull request #866 from myhloli/dev
fix(table): improve table image processing
7ca195e7 · Xiaomeng Zhao · GitHub · 7ec3a653 · 401dfa4e · 7ca195e7
Unverified Commit 7ca195e7 authored Nov 05, 2024 by Xiaomeng Zhao Committed by GitHub Nov 05, 2024
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

docs/README_Ubuntu_CUDA_Acceleration_zh_CN.md docs/README_Ubuntu_CUDA_Acceleration_zh_CN.md +0 -1

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

No files found.
--- a/docs/README_Ubuntu_CUDA_Acceleration_zh_CN.md
+++ b/docs/README_Ubuntu_CUDA_Acceleration_zh_CN.md
@@ -11,7 +11,6 @@ nvidia-smi
 注意:`CUDA Version` 显示的版本号应 >= 12.1，如显示的版本号小于12.1，请升级驱动
 ```plaintext
-```
 +---------------------------------------------------------------------------------------+
 | NVIDIA-SMI 537.34                 Driver Version: 537.34       CUDA Version: 12.2     |
 |-----------------------------------------+----------------------+----------------------+

--- 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(