[orientation] augment angle while training

[felixdittrich92]

This PR:

• augment randomly the angle while training (+-25 degree) (expand if not directly 0, 90, 180 or -90)
• the idea: the current orientation models works pretty well if the real angle of the page or crop is close to [0, 90, 180, -90] but has some lack with larger deviations like >+- 5 degree

@odulcy-mindee We should test it on the TF orientation model training runs i think this will improve the robustness a lot and then do the same again with the PT models (or we test it directly with the PT models)

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 95.88%. Comparing base (630d925) to head (8b93ef4).

Additional details and impacted files

@@           Coverage Diff           @@
##             main    #1567   +/-   ##
=======================================
  Coverage   95.88%   95.88%           
=======================================
  Files         163      163           
  Lines        7649     7649           
=======================================
  Hits         7334     7334           
  Misses        315      315           

Flag	Coverage Δ
unittests	95.88% <ø> (ø)

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[odulcy-mindee]

Will the expand arg not mess with the input size of the image ? Or will it be correctly resized before entering the training pipeline ?

[felixT2K]

Will the expand arg not mess with the input size of the image ? Or will it be correctly resized before entering the training pipeline ?

It's correctly resized before :)