mirror of
https://github.com/qurator-spk/eynollah.git
synced 2026-06-28 07:49:21 +02:00
models w/ multiple inputs yield a tuple for .input_shape
This commit is contained in:
parent
08946067ac
commit
45c92eada2
2 changed files with 4 additions and 2 deletions
|
|
@ -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 = []
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue