This repository contains all the source code for EE626-PRML Project titled : Gender Recognition from face images
- Kartikaeya Kumar (180108017)
- Swayam Bukharia (180108049)
- Adriraj Chaudhuri (180107003)
- Arpit Kumar Jain (180122009)
YouTube video link : ["EE626 : Gender Recognition from face images"]
Drive link
Download the dataset and place in $Root directory.
Three different model architectures were tried :
- Lightweight model (~2M parameters)
Training curve
- Medium weight model (~8M parameters)
Training curve
- Heavy weight model (~10M parameters)
Training curve
We also tried training the medium weight model with three different optimizers
- ADAgrad
- ADAM
- RMSprop
