assume_straight_pages vs. straighten_pages

[Adversarian]

The ocr_predictor class can take two kwargs called assume_straight_pages and straighten_pages. My understanding of the documentation is that straighten_pages=True should entail assume_straight_pages=False. However I wanted to ask to make sure when I need to use one or the other, or should I be using both when dealing with tilted/crooked documents (i.e. assume_straight_pages=False, straighten_pages=True) .

[felixdittrich92]

Hi @Adversarian 👋🏼,

straighten_pages=True should be used with assume_straight_pages=False in combination.
The specific of straighten_pages=True is, that we try to estimate the orientation angle from the document, rotate it counter clockwise to "straighten" the image and atm. process it again with the straightened image.
Without this so only assume_straight_pages=False we don't try to straighten it and process it as is (output geometry are also polygons).

Does this help ? :)

[Adversarian]

Thanks for your response. However I still don't understand what straighten_pages=False, assume_straight_pages=False does. If we're not assuming that the pages are straight, yet we're also not straightening the pages before re-running the pipeline on the image then what is it exactly that assume_straight_pages=False does?

[felixdittrich92]

In short:

with assume_straight_pages=False under the hood it's handled as quad polygons [[x1, y1], [x2, y2], [x3, y3], [x4, y4]] instead of boxes [xmin, ymin, xmax, ymax] .. this starts by extracting the coords (detection postprocessor) from the detection model results .. and goes trough the whole pipeline

So the pipeline is the same but with assume_straight_pages=False we have also to take care of rotations and compute "on polygons" instead of boxes

• additional to support this we plug in (now on main branch) 2 tiny models for general crop and page orientation classification

Try a rotated document with assume_straight_pages=False and assume_straight_pages=True to get a visual feedback.

A bit clearer ? :)

[Adversarian]

That clears everything. Thank you.