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

refactoring

Browse Source
This commit is contained in:
Matthias Bisping 2022-04-12 16:34:00 +02:00
parent 35c5b15e32
commit e82a81f5c8
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,6 +1,6 @@
import io
from functools import partial, lru_cache
from itertools import chain, starmap
from itertools import chain, starmap, filterfalse
from operator import itemgetter, truth
import fitz
@ -72,7 +72,7 @@ def get_metadata_for_images_on_page(page: fitz.Page):
image_infos = get_image_infos(page)
metadata = map(get_image_metadata, image_infos)
metadata = validate_coords_and_passthrough(metadata)
metadata = filter(tiny, metadata)
metadata = filterfalse(tiny, metadata)
metadata = validate_size_and_passthrough(metadata)
metadata = map(partial(merge, get_page_metadata(page)), metadata)
@ -114,7 +114,7 @@ def get_image_metadata(image_info):
def tiny(metadata):
return metadata[Info.WIDTH] * metadata[Info.HEIGHT]
return metadata[Info.WIDTH] * metadata[Info.HEIGHT] <= 4
def get_page_metadata(page):