Marmelator/image-classification-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 29 Wiki Activity

return -> yield from

Browse Source
This commit is contained in:
Matthias Bisping 2022-04-19 14:29:02 +02:00
parent 2f37d20f68
commit 5df3d1e85b
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
image_prediction/image_extractor/extractors/parsable.py
View File

@ -58,7 +58,7 @@ def extract_pages(doc, page_range):
page_range = range(page_range.start + 1, page_range.stop + 1) page_range = range(page_range.start + 1, page_range.stop + 1)
pages = map(doc.load_page, page_range) pages = map(doc.load_page, page_range)
return pages yield from pages
@lru_cache(maxsize=None) @lru_cache(maxsize=None)
@ -67,7 +67,7 @@ def get_images_on_page(doc, page: fitz.Page):
xrefs = map(itemgetter("xref"), image_infos) xrefs = map(itemgetter("xref"), image_infos)
images = map(partial(xref_to_image, doc), xrefs) images = map(partial(xref_to_image, doc), xrefs)
return images yield from images
def get_metadata_for_images_on_page(doc, page: fitz.Page): def get_metadata_for_images_on_page(doc, page: fitz.Page):
@ -118,7 +118,7 @@ def validate_coords_and_passthrough(metadata):
def filter_out_tiny_images(metadata): def filter_out_tiny_images(metadata):
return filterfalse(tiny, metadata) yield from filterfalse(tiny, metadata)
def validate_size_and_passthrough(metadata): def validate_size_and_passthrough(metadata):
@ -126,7 +126,7 @@ def validate_size_and_passthrough(metadata):
def add_page_metadata(page, metadata): def add_page_metadata(page, metadata):
return map(partial(merge, get_page_metadata(page)), metadata) yield from map(partial(merge, get_page_metadata(page)), metadata)
def add_alpha_channel_info(doc, page, metadata): def add_alpha_channel_info(doc, page, metadata):
@ -139,7 +139,7 @@ def add_alpha_channel_info(doc, page, metadata):
metadata = starmap(merge, zip(page_to_alpha_mapping_per_image(page), metadata)) metadata = starmap(merge, zip(page_to_alpha_mapping_per_image(page), metadata))
return metadata yield from metadata
@lru_cache(maxsize=None) @lru_cache(maxsize=None)