mirrors
/
PaddleOCR
mirror of https://github.com/PaddlePaddle/PaddleOCR.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix fitz camelCase deprecation and .PDF not being recognized as pdf file (#10181) (#10707) 

* Fix fitz camelCase deprecation and .PDF not being recognized as pdf file

* refactor get_image_file_list function
pull/10709/head
itasli 2023-08-23 09:46:25 +02:00 committed by GitHub
parent b162237e84
commit 9ef797a2d2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
ppocr/utils/utility.py
View File

@ -62,7 +62,6 @@ def get_image_file_list(img_file):
if img_file is None or not os.path.exists(img_file): if img_file is None or not os.path.exists(img_file):
raise Exception("not found any img file in {}".format(img_file)) raise Exception("not found any img file in {}".format(img_file))
img_end = {'jpg', 'bmp', 'png', 'jpeg', 'rgb', 'tif', 'tiff', 'gif', 'pdf'}
if os.path.isfile(img_file) and _check_image_file(img_file): if os.path.isfile(img_file) and _check_image_file(img_file):
imgs_lists.append(img_file) imgs_lists.append(img_file)
elif os.path.isdir(img_file): elif os.path.isdir(img_file):
@ -96,7 +95,7 @@ def alpha_to_color(img, alpha_color=(255, 255, 255)):
return img return img
def check_and_read(img_path): def check_and_read(img_path):
if os.path.basename(img_path)[-3:] in ['gif', 'GIF']: if os.path.basename(img_path)[-3:].lower() == 'gif':
gif = cv2.VideoCapture(img_path) gif = cv2.VideoCapture(img_path)
ret, frame = gif.read() ret, frame = gif.read()
if not ret: if not ret:
@ -107,19 +106,19 @@ def check_and_read(img_path):
frame = cv2.cvtColor(frame, cv2.COLOR_GRAY2RGB) frame = cv2.cvtColor(frame, cv2.COLOR_GRAY2RGB)
imgvalue = frame[:, :, ::-1] imgvalue = frame[:, :, ::-1]
return imgvalue, True, False return imgvalue, True, False
elif os.path.basename(img_path)[-3:] in ['pdf']: elif os.path.basename(img_path)[-3:].lower() == 'pdf':
import fitz import fitz
from PIL import Image from PIL import Image
imgs = [] imgs = []
with fitz.open(img_path) as pdf: with fitz.open(img_path) as pdf:
for pg in range(0, pdf.pageCount): for pg in range(0, pdf.page_count):
page = pdf[pg] page = pdf[pg]
mat = fitz.Matrix(2, 2) mat = fitz.Matrix(2, 2)
pm = page.getPixmap(matrix=mat, alpha=False) pm = page.get_pixmap(matrix=mat, alpha=False)
# if width or height > 2000 pixels, don't enlarge the image # if width or height > 2000 pixels, don't enlarge the image
if pm.width > 2000 or pm.height > 2000: if pm.width > 2000 or pm.height > 2000:
pm = page.getPixmap(matrix=fitz.Matrix(1, 1), alpha=False) pm = page.get_pixmap(matrix=fitz.Matrix(1, 1), alpha=False)
img = Image.frombytes("RGB", [pm.width, pm.height], pm.samples) img = Image.frombytes("RGB", [pm.width, pm.height], pm.samples)
img = cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR) img = cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR)