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