File size: 473 Bytes
c8a32e7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
from typing import List
from marker.schema.page import Page
def get_image_filename(page: Page, image_idx):
return f"{page.pnum}_image_{image_idx}.png"
def images_to_dict(pages: List[Page]):
images = {}
for page in pages:
if page.images is None:
continue
for image_idx, image in enumerate(page.images):
image_filename = get_image_filename(page, image_idx)
images[image_filename] = image
return images
|