From a3a7ab034b18bfacd34596f5399fac618eb1cc6c Mon Sep 17 00:00:00 2001
From: Wenjie Zhang <xerrors@163.com>
Date: Wed, 11 Sep 2024 01:08:37 +0800
Subject: [PATCH] trans cv2 to PIL

---
 src/plugins/pdf2txt.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/plugins/pdf2txt.py b/src/plugins/pdf2txt.py
index ae64ea39..bb9c4d07 100644
--- a/src/plugins/pdf2txt.py
+++ b/src/plugins/pdf2txt.py
@@ -1,11 +1,12 @@
 import os
 import uuid
 import fitz  # fitz就是pip install PyMuPDF
-import cv2
+from PIL import Image
 from copy import deepcopy
 from tqdm import tqdm
 from argparse import ArgumentParser
 from src.utils import logger
+import numpy as np  # Added import for numpy
 
 GOLBAL_STATE = {}
 
@@ -45,7 +46,8 @@ def pdf2txt(pdf_path, return_text=False):
 
     text = []
     for img_name in tqdm(imgs, desc='to txt', ncols=100):
-        img = cv2.imread(img_name)
+        img = Image.open(img_name)
+        img = np.array(img)  # 如果需要将图像转换为numpy数组
         result = table_engine(img)
         GOLBAL_STATE[task_id]['progress'] += 1