From 7b937d58b83a4367606ec02c45efef393978fdd6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=B5=B5=E5=B0=8F=E8=92=99?= Date: Mon, 15 Apr 2024 22:15:30 +0800 Subject: [PATCH] =?UTF-8?q?fix=20=E5=8F=82=E6=95=B0=E5=90=8D=E7=A7=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- magic_pdf/libs/pdf_image_tools.py | 4 ++-- magic_pdf/pipe/UNIPipe.py | 8 ++++---- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/magic_pdf/libs/pdf_image_tools.py b/magic_pdf/libs/pdf_image_tools.py index abc92ea9..1be921f5 100644 --- a/magic_pdf/libs/pdf_image_tools.py +++ b/magic_pdf/libs/pdf_image_tools.py @@ -1,4 +1,4 @@ - +from magic_pdf.io.AbsReaderWriter import AbsReaderWriter from magic_pdf.libs.commons import fitz from loguru import logger from magic_pdf.libs.commons import join_path @@ -28,7 +28,7 @@ def cut_image(bbox: tuple, page_num: int, page: fitz.Page, return_path, imageWri byte_data = pix.tobytes(output='jpeg', jpg_quality=95) - imageWriter.write(content=byte_data, path=img_hash256_path, mode="binary") + imageWriter.write(byte_data, img_hash256_path, AbsReaderWriter.MODE_BIN) return img_hash256_path diff --git a/magic_pdf/pipe/UNIPipe.py b/magic_pdf/pipe/UNIPipe.py index 519b1a87..c05f2fd6 100644 --- a/magic_pdf/pipe/UNIPipe.py +++ b/magic_pdf/pipe/UNIPipe.py @@ -105,8 +105,8 @@ if __name__ == '__main__': pdf_file_path = r"linshixuqiu\25536-00.pdf" model_file_path = r"linshixuqiu\25536-00.json" - pdf_bytes = drw.read(path=pdf_file_path, mode=AbsReaderWriter.MODE_BIN) - model_json_txt = drw.read(path=model_file_path, mode=AbsReaderWriter.MODE_TXT) + pdf_bytes = drw.read(pdf_file_path, AbsReaderWriter.MODE_BIN) + model_json_txt = drw.read(model_file_path, AbsReaderWriter.MODE_TXT) pdf_type = UNIPipe.classify(pdf_bytes) logger.info(f"pdf_type is {pdf_type}") @@ -122,5 +122,5 @@ if __name__ == '__main__': md_content = pipe.mk_markdown(pdf_mid_data, "imgs") md_writer = DiskReaderWriter(write_path) - md_writer.write(content=md_content, path="25536-00.md", mode=AbsReaderWriter.MODE_TXT) - md_writer.write(content=json.dumps(JsonCompressor.decompress_json(pdf_mid_data), ensure_ascii=False, indent=4), path="25536-00.json", mode=AbsReaderWriter.MODE_TXT) + md_writer.write(md_content, "25536-00.md", AbsReaderWriter.MODE_TXT) + md_writer.write(json.dumps(JsonCompressor.decompress_json(pdf_mid_data), ensure_ascii=False, indent=4), "25536-00.json", AbsReaderWriter.MODE_TXT)