Estimate age, gender, and ethnicity from facial images using a ResNet model.
This project utilizes the UTK Faces dataset, by cropping the facial region, to train two ResNet models for facial attribute estimation:
- ResNet-50 for age estimation
- ResNet-18 for gender and ethnicity classification
Dataset link: UTK Faces
Pretrained models are included in this repository for direct inference.
If running notebooks in Google Colab, set system = "colab"
in the Utils section.
For local execution, set system = "local"
.