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

removed obsolete imports

Browse Source
This commit is contained in:
Matthias Bisping 2022-04-04 21:50:10 +02:00
parent 8f61c4cba2
commit ce69f7d160
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,11 +1,10 @@
import io
from itertools import chain, starmap
from operator import itemgetter, __and__, truth
from operator import itemgetter, truth
import fitz
from PIL import Image
from funcy import rcompose, compose, curry
from iteration_utilities import starfilter
from tqdm import tqdm
from image_prediction.image_extractor.extractor import ImageExtractor, ImageMetadataPair
@ -49,7 +48,7 @@ class ParsablePDFImageExtractor(ImageExtractor):
images = map(load_image_from_xref, xrefs)
metadata = map(format_metadata, image_infos)
return starmap(ImageMetadataPair, filter(compose(all, curry(map)(truth)), zip(images, metadata)))
return starmap(ImageMetadataPair, filter(compose(all, curry(map)(truth)), zip(images, metadata)))
def extract(self, pdf: bytes):
self.doc = fitz.Document(stream=pdf)