kris/ForcePilot
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

trans cv2 to PIL

Browse Source
This commit is contained in:
Wenjie Zhang 2024-09-11 01:08:37 +08:00
parent 152bfbf49f
commit a3a7ab034b
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/plugins/pdf2txt.py
View File

@ -1,11 +1,12 @@
import os import os
import uuid import uuid
import fitz # fitz就是pip install PyMuPDF import fitz # fitz就是pip install PyMuPDF
import cv2 from PIL import Image
from copy import deepcopy from copy import deepcopy
from tqdm import tqdm from tqdm import tqdm
from argparse import ArgumentParser from argparse import ArgumentParser
from src.utils import logger from src.utils import logger
import numpy as np # Added import for numpy
GOLBAL_STATE = {} GOLBAL_STATE = {}
@ -45,7 +46,8 @@ def pdf2txt(pdf_path, return_text=False):
text = [] text = []
for img_name in tqdm(imgs, desc='to txt', ncols=100): 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) result = table_engine(img)
GOLBAL_STATE[task_id]['progress'] += 1 GOLBAL_STATE[task_id]['progress'] += 1