mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-08-02 15:32:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

Updated examples to use get_image instead of element.image

Browse Source 
Signed-off-by: Shubham Gupta <26436285+sh-gupta@users.noreply.github.com>
This commit is contained in:
Shubham Gupta 2024-11-18 19:00:00 +01:00
parent 7a79047af5
commit 8c0be69408
2 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/examples/develop_picture_enrichment.py
View File

@ -39,7 +39,7 @@ class ExamplePictureClassifierEnrichmentModel(BaseEnrichmentModel):
assert isinstance(element, PictureItem) assert isinstance(element, PictureItem)
# uncomment this to interactively visualize the image # uncomment this to interactively visualize the image
# element.image.pil_image.show() # element.get_image(doc).show()
element.annotations.append( element.annotations.append(
PictureClassificationData( PictureClassificationData(

5
docs/examples/export_figures.py
View File

@ -28,7 +28,6 @@ def main():
pipeline_options = PdfPipelineOptions() pipeline_options = PdfPipelineOptions()
pipeline_options.images_scale = IMAGE_RESOLUTION_SCALE pipeline_options.images_scale = IMAGE_RESOLUTION_SCALE
pipeline_options.generate_page_images = True pipeline_options.generate_page_images = True
pipeline_options.generate_table_images = True
pipeline_options.generate_picture_images = True pipeline_options.generate_picture_images = True
doc_converter = DocumentConverter( doc_converter = DocumentConverter(
@ -61,7 +60,7 @@ def main():
output_dir / f"{doc_filename}-table-{table_counter}.png" output_dir / f"{doc_filename}-table-{table_counter}.png"
) )
with element_image_filename.open("wb") as fp: with element_image_filename.open("wb") as fp:
element.image.pil_image.save(fp, "PNG") element.get_image(conv_res.document).save(fp, "PNG")
if isinstance(element, PictureItem): if isinstance(element, PictureItem):
picture_counter += 1 picture_counter += 1
@ -69,7 +68,7 @@ def main():
output_dir / f"{doc_filename}-picture-{picture_counter}.png" output_dir / f"{doc_filename}-picture-{picture_counter}.png"
) )
with element_image_filename.open("wb") as fp: with element_image_filename.open("wb") as fp:
element.image.pil_image.save(fp, "PNG") element.get_image(conv_res.document).save(fp, "PNG")
# Save markdown with embedded pictures # Save markdown with embedded pictures
content_md = conv_res.document.export_to_markdown(image_mode=ImageRefMode.EMBEDDED) content_md = conv_res.document.export_to_markdown(image_mode=ImageRefMode.EMBEDDED)