models w/ multiple inputs yield a tuple for .input_shape

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

View file

@ -166,7 +166,8 @@ class Eynollah_ocr(Eynollah):
page_tree: ET.ElementTree,
page_ns,
) -> 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 = []
cropped_lines_rgb = []