From 53ccd5a688a40e1300c7bb3d34fd42bd7dbfadbc Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=B5=B5=E5=B0=8F=E8=92=99?= Date: Fri, 28 Jun 2024 12:48:31 +0800 Subject: [PATCH] fix: add try import opencv-python and Pillow --- magic_pdf/model/doc_analyze_by_custom_model.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/magic_pdf/model/doc_analyze_by_custom_model.py b/magic_pdf/model/doc_analyze_by_custom_model.py index 0415502d..770c0e48 100644 --- a/magic_pdf/model/doc_analyze_by_custom_model.py +++ b/magic_pdf/model/doc_analyze_by_custom_model.py @@ -1,9 +1,6 @@ import fitz -import cv2 -from PIL import Image import numpy as np from loguru import logger - from magic_pdf.model.model_list import MODEL import magic_pdf.model as model_config @@ -23,6 +20,11 @@ def remove_duplicates_dicts(lst): def load_images_from_pdf(pdf_bytes: bytes, dpi=200) -> list: + try: + import cv2 + from PIL import Image + except ImportError: + logger.error("opencv-python and Pillow are not installed, please install by pip.") images = [] with fitz.open("pdf", pdf_bytes) as doc: for index in range(0, doc.page_count):