qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2026-06-16 09:59:13 +02:00
Code Issues Projects Releases Wiki Activity

models w/ multiple inputs yield a tuple for .input_shape

Browse source
This commit is contained in:
Robert Sachunsky 2026-06-12 14:55:46 +02:00
parent 08946067ac
commit 45c92eada2
2 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/eynollah/eynollah_ocr.py
View file

@ -166,7 +166,8 @@ class Eynollah_ocr(Eynollah):
page_tree: ET.ElementTree, page_tree: ET.ElementTree,
page_ns, page_ns,
) -> EynollahOcrResult: ) -> EynollahOcrResult:
_, image_height, image_width, _ = self.model_zoo.get('ocr').input_shape input_shape, _ = self.model_zoo.get('ocr').input_shape
_, image_height, image_width, _ = input_shape
total_bb_coordinates = [] total_bb_coordinates = []
cropped_lines_rgb = [] cropped_lines_rgb = []

3
tests/test_model_zoo.py
View file

@ -23,6 +23,7 @@ def test_cnnrnnocr1(
model = model_zoo.get('ocr') model = model_zoo.get('ocr')
assert isinstance(model, Predictor) assert isinstance(model, Predictor)
shape = model.input_shape shape = model.input_shape
assert len(shape) == 4 assert len(shape) == 2
assert len(shape[0]) == 4
except ImportError: except ImportError:
pass pass