Skip to content

Age and Gender detection ,Keras implementation using CNN

Notifications You must be signed in to change notification settings

charvi-077/AI-Project

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 

Repository files navigation

Age and Gender Estimation

This is a Keras implementation of a CNN for estimating age and gender from a face image . In training, the IMDB-WIKI dataset is used.

Dependencies

  • Python 3
  • Keras2.0+
  • scipy, numpy, Pandas, tqdm
  • OpenCV

Tested on:

  • CPU ,GPU:k80 (colab)
  • Python 3.6.0, Keras 2.0.2, Tensorflow 1.0.0

Download the dataset

The dataset is downloaded and extracted to the data directory.

!wget https://data.vision.ee.ethz.ch/cvl/rrothe/imdb-wiki/static/imdb_crop.tar

Train network

Train the network using the training data created above.

Plot training curves from history file

python3 plot_history.py --input models/history_16_8.h5 
Results with data

The val_loss was 4.7

Network architecture

Here the Wide Residual Network (WideResNet) is trained from scratch. The implementation of the WideResNet; two classification layers (for age and gender estimation) are added on the top of the WideResNet.

While age and gender are independently estimated by different two CNNs ,they are simultaneously estimated using a single CNN.

Results

Trained on imdb, tested on imdb

Refrences

https://www.openu.ac.il/home/hassner/projects/cnn_agegender

About

Age and Gender detection ,Keras implementation using CNN

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published