We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 1cea7d8 commit a659425Copy full SHA for a659425
doctr/models/detection/zoo.py
@@ -61,6 +61,7 @@ def _predictor(arch: Any, pretrained: bool, assume_straight_pages: bool = True,
61
62
_model = arch
63
_model.assume_straight_pages = assume_straight_pages
64
+ _model.postprocessor.assume_straight_pages = assume_straight_pages
65
66
kwargs.pop("pretrained_backbone", None)
67
0 commit comments