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-21 19:48:05 +02:00
parent 72e785e3e3
commit 6c10b55ff8
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
image_prediction/pipeline.py
View File

@ -35,9 +35,6 @@ class Pipeline:
def join_prediction_and_metadata(prd, mdt): def join_prediction_and_metadata(prd, mdt):
return {"classification": prd, **mdt} return {"classification": prd, **mdt}
def process_batch(batch):
classifications, metadata = juxt(left, right)(batch)
return starmap(join_prediction_and_metadata, zip(classifications, metadata))
# -------- # --------
# -- -- -- -- # -- -- -- --
@ -55,7 +52,9 @@ class Pipeline:
self.pipe = rcompose( self.pipe = rcompose(
extractor, extractor,
batcher, batcher,
lift(process_batch), lift(juxt(left, right)),
starlift(zip),
lift(starlift(join_prediction_and_metadata)),
chain.from_iterable, chain.from_iterable,
formatter, formatter,
) )