GylbarD commited on
Commit
88336ce
·
verified ·
1 Parent(s): a31c578

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +23 -3
README.md CHANGED
@@ -1,3 +1,23 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ metrics:
4
+ - accuracy 99.51%
5
+ tags:
6
+ - OrientationDetection
7
+ - TextOrientation
8
+ ---
9
+
10
+ Updated CNN .keras model with input layer 256*256 RGB, and 1 output node ranging from 0 (no rotation) to 1 (180 degrees).
11
+
12
+ Its accuracy on the dataset in the link below is 99.51%, and the AUC (Area Under Curve) metric is 0.9990.
13
+ https://drive.google.com/file/d/1WENDw-Ggwy32KIdMlMmuF6NcO0gH3mC8/view?usp=sharing
14
+
15
+ It only recognizes 0 vs 180 degrees as there's another robust way to distinguish +- 90 from 0 and 180 degrees (another model under UTSC DSU).
16
+ Additionally, the dataset only contains upright images because the training program generates the 180-degrees counterparts on the fly.
17
+ As you will see, the model is only trained on text slices resulted from my proprietary preprocessing methods
18
+ (old version already on GitHub https://github.com/digitalutsc/ml_image_tool, but I will update it soon),
19
+ So directly feeding the page images to the model may not work well.
20
+
21
+ I used MobileNetV2 with imagenet weights as the backbone, and the accuracy proves the method to be very successful.
22
+
23
+ ![hist_outputs](https://cdn-uploads.huggingface.co/production/uploads/67fc828a84224b6e5a42d981/uQWaG8Nsq5Xx6IcatFzEm.png)