Code for building the race and gender prediction model for my final project. Used to (hopefully) yield faster predictions.
UTK Face Dataset: https://www.kaggle.com/datasets/jangedoo/utkface-new
Keras API: https://keras.io/guides/functional_api/
Code Reference: https://github.com/rodrigobressan/keras-multi-output-model-utk-face?tab=readme-ov-file
- Clone repository
- Add dataset to a folder called /UTKFace in the base directory
- Run jupyter notebook