mirror of
https://github.com/DS4SD/docling.git
synced 2025-07-27 12:34:22 +00:00
62 lines
1.9 KiB
Python
62 lines
1.9 KiB
Python
import logging
|
|
import time
|
|
from pathlib import Path
|
|
|
|
from docling.datamodel.base_models import FigureElement, InputFormat, Table
|
|
from docling.datamodel.pipeline_options import PdfPipelineOptions
|
|
from docling.document_converter import DocumentConverter, PdfFormatOption
|
|
|
|
_log = logging.getLogger(__name__)
|
|
|
|
IMAGE_RESOLUTION_SCALE = 2.0
|
|
|
|
|
|
def main():
|
|
logging.basicConfig(level=logging.INFO)
|
|
|
|
input_doc_path = Path("./tests/data/2206.01062.pdf")
|
|
output_dir = Path("./scratch")
|
|
|
|
# Important: For operating with page images, we must keep them, otherwise the DocumentConverter
|
|
# will destroy them for cleaning up memory.
|
|
# This is done by setting AssembleOptions.images_scale, which also defines the scale of images.
|
|
# scale=1 correspond of a standard 72 DPI image
|
|
pipeline_options = PdfPipelineOptions()
|
|
pipeline_options.images_scale = IMAGE_RESOLUTION_SCALE
|
|
|
|
doc_converter = DocumentConverter(
|
|
format_options={
|
|
InputFormat.PDF: PdfFormatOption(pipeline_options=pipeline_options)
|
|
}
|
|
)
|
|
|
|
start_time = time.time()
|
|
|
|
conv_res = doc_converter.convert(input_doc_path)
|
|
|
|
output_dir.mkdir(parents=True, exist_ok=True)
|
|
doc_filename = conv_res.input.file.stem
|
|
|
|
# Export page images
|
|
for page in conv_res.pages:
|
|
page_no = page.page_no + 1
|
|
page_image_filename = output_dir / f"{doc_filename}-{page_no}.png"
|
|
with page_image_filename.open("wb") as fp:
|
|
page.image.save(fp, format="PNG")
|
|
|
|
# Export figures and tables
|
|
for element, image in conv_res.render_element_images(
|
|
element_types=(FigureElement, Table)
|
|
):
|
|
element_image_filename = output_dir / f"{doc_filename}-element-{element.id}.png"
|
|
with element_image_filename.open("wb") as fp:
|
|
image.save(fp, "PNG")
|
|
|
|
end_time = time.time() - start_time
|
|
|
|
_log.info(f"Document converted and figures exported in {end_time:.2f} seconds.")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|