From d48005a2e1836fe3b71b116517ffd51eb13ff094 Mon Sep 17 00:00:00 2001 From: Bruno Picinin Cardoso Date: Fri, 17 Jan 2025 18:13:38 -0300 Subject: [PATCH] Fix bug caused by google vision response including vertices with only 'x' or 'y' --- .../core_steps/models/foundation/google_vision_ocr/v1.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/inference/core/workflows/core_steps/models/foundation/google_vision_ocr/v1.py b/inference/core/workflows/core_steps/models/foundation/google_vision_ocr/v1.py index 2fd8431f65..27ff10406c 100644 --- a/inference/core/workflows/core_steps/models/foundation/google_vision_ocr/v1.py +++ b/inference/core/workflows/core_steps/models/foundation/google_vision_ocr/v1.py @@ -162,10 +162,10 @@ def run( for block in page["blocks"]: # Get bounding box coordinates box = block["boundingBox"]["vertices"] - x_min = min(v["x"] for v in box) - y_min = min(v["y"] for v in box) - x_max = max(v["x"] for v in box) - y_max = max(v["y"] for v in box) + x_min = min(v.get("x", 0) for v in box) + y_min = min(v.get("y", 0) for v in box) + x_max = max(v.get("x", 0) for v in box) + y_max = max(v.get("y", 0) for v in box) xyxy.append([x_min, y_min, x_max, y_max]) # Only DOCUMENT_TEXT_DETECTION provides confidence score, use 1.0 otherwise