fix assume_straight_pages with custom loaded models

felixdittrich92 · Jan 13, 2025 · b55d58f · b55d58f
1 parent aa20687
commit b55d58f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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"])