[New Model] – GyroScope: rotates images correctly

[New Model] - GyroScope: rotates images correctly

Hey there!
I have made a new model: https://huggingface.co/LH-Tech-AI/GyroScope

So, you just input a image (rotated by 0°, 90°, 180° or 270°) and the model corrects the rotation to make it correct.

Example:

https://preview.redd.it/kceygtv0mkug1.png?width=1012&format=png&auto=webp&s=562e1454a3be26b79ca9a53960981a71640ea9dc

I tested it with lots of photos - and it almost always was correct :D

Final accuracy after 12 epochs of training (~4h on single T4):

Metric Value
Overall Val Accuracy 79.81%%
Per-class: 0° (upright) 79.8%
Per-class: 90° CCW 80.1%
Per-class: 180° 79.4%
Per-class: 270° CCW 79.8%
Training Epochs 12
Training Time ~4h (Kaggle T4 GPU)

Tell me what you think about it :-)

submitted by /u/LH-Tech_AI
[link] [comments]

Leave a Comment

Your email address will not be published. Required fields are marked *

Scroll to Top