Skip to content

Commit 5a45ad4

Browse files
added age gender test results + model description
1 parent 5aa3329 commit 5a45ad4

File tree

1 file changed

+25
-1
lines changed

1 file changed

+25
-1
lines changed

README.md

Lines changed: 25 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -803,4 +803,28 @@ The face expression recognition model is lightweight, fast and provides reasonab
803803

804804
## Age and Gender Recognition Model
805805

806-
TBD...
806+
The age and gender recognition model is a multitask network, which employs a feature extraction layer, an age regression layer and a gender classifier. The model has a size of roughly 420kb and the feature extractor employs a tinier but very similar architecture to Xception.
807+
808+
This model has been trained and tested on the following databases with an 80/20 train/test split each: UTK, FGNET, Chalearn, Wiki, IMDB*, CACD*, MegaAge, MegaAge-Asian. The `*` indicates, that these databases have been algorithmically cleaned up, since the initial databases are very noisy.
809+
810+
### Total Test Results
811+
812+
Total MAE (Mean Age Error): **4.54**
813+
814+
Total Gender Accuracy: **95%**
815+
816+
### Test results for each database
817+
818+
The `-` indicates, that there are no gender labels available for these databases.
819+
820+
Database | UTK | FGNET | Chalearn | Wiki | IMDB* | CACD* | MegaAge | MegaAge-Asian |
821+
----------------|-------:|------:|---------:|-----:|------:|------:|--------:|--------------:|
822+
MAE | 5.25 | 4.23 | 6.24 | 6.54 | 3.63 | 3.20 | 6.23 | 4.21 |
823+
Gender Accuracy | 0.93 | - | 0.94 | 0.95 | - | 0.97 | - | - |
824+
825+
### Test results for different age category groups
826+
827+
Age Range | 0 - 3 | 4 - 8 | 9 - 18 | 19 - 28 | 29 - 40 | 41 - 60 | 60 - 80 | 80+ |
828+
----------------|-------:|------:|-------:|--------:|--------:|--------:|--------:|--------:|
829+
MAE | 1.52 | 3.06 | 4.82 | 4.99 | 5.43 | 4.94 | 6.17 | 9.91 |
830+
Gender Accuracy | 0.69 | 0.80 | 0.88 | 0.96 | 0.97 | 0.97 | 0.96 | 0.9 |

0 commit comments

Comments
 (0)