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-07 17:28:25 +02:00
parent 3e882dc247
commit 2fcb0bd149
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
test/unit_tests/image_stitcher_test.py
View File

@ -8,7 +8,7 @@ import fpdf
import numpy as np import numpy as np
import pytest import pytest
from PIL import Image from PIL import Image
from funcy import merge, second, compose, rpartial, juxt from funcy import merge, second, compose, rpartial, juxt, rest, first
from image_prediction.estimator.preprocessor.utils import image_to_normalized_tensor from image_prediction.estimator.preprocessor.utils import image_to_normalized_tensor
from image_prediction.image_extractor.extractor import ImageMetadataPair from image_prediction.image_extractor.extractor import ImageMetadataPair
@ -127,8 +127,8 @@ class Stitcher:
def merge_group(group): def merge_group(group):
def f(pairs):
current_pair = pairs.pop(0) def merge_with(current_pair, pairs):
to_remove = [] to_remove = []
@ -142,7 +142,7 @@ def merge_group(group):
pairs = list(group) pairs = list(group)
while True: while True:
new_pairs = f(deepcopy(pairs)) new_pairs = merge_with(*juxt(first, rest)(pairs))
if len(new_pairs) == len(pairs): if len(new_pairs) == len(pairs):
break break
pairs = new_pairs pairs = new_pairs