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

Add to-do

Browse Source
This commit is contained in:
Matthias Bisping 2023-02-07 10:18:53 +01:00
parent 3cea4dad2d
commit 092069221a
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -193,7 +193,7 @@ def has_alpha_channel(doc: Doc, xref: int):
maybe_image = load_image_handle_from_xref(doc, xref)
maybe_smask = maybe_image["smask"] if maybe_image else None
if maybe_smask: # Use monad
if maybe_smask: # TODO: Use monad.
return any([doc.extract_image(maybe_smask) is not None, bool(fitz.Pixmap(doc, maybe_smask).alpha)])
else:
try:
@ -261,7 +261,7 @@ def normalize_channels(array: np.ndarray):
@lru_cache(maxsize=None)
def load_image_handle_from_xref(doc: Doc, xref: int) -> Union[dict, None]: # TODO: use Monad
def load_image_handle_from_xref(doc: Doc, xref: int) -> Union[dict, None]: # TODO: Use monad.
return doc.extract_image(xref)