From b55d58f2c40539221a1269ec9b8a1d8d89706390 Mon Sep 17 00:00:00 2001 From: felix Date: Mon, 13 Jan 2025 08:21:23 +0100 Subject: [PATCH] fix assume_straight_pages with custom loaded models --- onnxtr/models/detection/zoo.py | 1 + 1 file changed, 1 insertion(+) diff --git a/onnxtr/models/detection/zoo.py b/onnxtr/models/detection/zoo.py index ccc5665..9c921c5 100644 --- a/onnxtr/models/detection/zoo.py +++ b/onnxtr/models/detection/zoo.py @@ -44,6 +44,7 @@ def _predictor( raise ValueError(f"unknown architecture: {type(arch)}") _model = arch + _model.assume_straight_pages = assume_straight_pages _model.postprocessor.assume_straight_pages = assume_straight_pages kwargs["mean"] = kwargs.get("mean", _model.cfg["mean"])