[references] TF / PT crop & document orientation classifier train scripts

[felixdittrich92]

This PR:

• Add training scripts (TF & PT) for the crop orientation classifier and additional for general document orientation classification (needed for 0.9.0 (similar to the crop_orientation_predictor) to generally recognize the documents rotation angle correctly before we compute the angle from the bin_map - which works only for rotated documents in the range ~ -60-60 well)
• Add a simple dataset (prefilled with 0 degree targets - to match the AbstractDataset checks)

Any feedback is welcome :)

• Tests
• Optimization (input_size tests left) - for crop orientation (32x32 is enough and works well) / for document orientation (256x256 seems to be enough)
• Clean up
• Readme

Dummy runs (mobilenet_v3_small)

crop orientation (800k train / 200k val)
Validation loss decreased 0.372947 --> 0.355291: saving state...
Epoch 5/30 - Validation loss: 0.355291 (Acc: 81.51%)

doc orientation (only 4k train / 1k val)
Validation loss decreased 0.559973 --> 0.481535: saving state...
Epoch 14/30 - Validation loss: 0.481535 (Acc: 79.22%)

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Comparison is base (bcf3cd4) 95.76% compared to head (8a65bd3) 95.84%.
Report is 2 commits behind head on main.

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #1432      +/-   ##
==========================================
+ Coverage   95.76%   95.84%   +0.07%     
==========================================
  Files         155      162       +7     
  Lines        6941     7095     +154     
==========================================
+ Hits         6647     6800     +153     
- Misses        294      295       +1     

Flag	Coverage Δ
unittests	95.84% <100.00%> (+0.07%)	⬆️

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]

Code seems good 👍

[felixdittrich92]

@odulcy-mindee So this part could be merged into 0.8.0 and the rest of this task is for 0.9.0 :)
Are you fine with the additional OrientationDataset if was thinking on putting it into references/classification/utils but i think the dataset folder is a better fit 😅

[odulcy-mindee]

Are you fine with the additional OrientationDataset if was thinking on putting it into references/classification/utils but i think the dataset folder is a better fit 😅
@felixdittrich92 yeah, that makes more sense 👍

I added one more suggestion.
For another PR: is it possible to have only one train_{framework}.py for both purposes ? Otherwise, maybe it makes more sense to rename the other scripts train_{framework}_character.py or something else ?

[odulcy-mindee]

Thank you @felixdittrich92 ! 🚀

[felixdittrich92]

Are you fine with the additional OrientationDataset if was thinking on putting it into references/classification/utils but i think the dataset folder is a better fit 😅
@felixdittrich92 yeah, that makes more sense 👍

I added one more suggestion. For another PR: is it possible to have only one train_{framework}.py for both purposes ? Otherwise, maybe it makes more sense to rename the other scripts train_{framework}_character.py or something else ?

I totally missed this 😅
Merging both into one makes no sense in my mind i think this would be confusing
But renaming the other one with suffix character would be fine 👍

[odulcy-mindee]

But renaming the other one with suffix character would be fine 👍

Ok, we can do that in another PR