Skip to content

Latest commit

 

History

1 Commit

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Age and Gender prediction

Inference python script for several age and gender estimation neural network models.

Requirements

  • python3.6
  • numpy==1.13.0
  • scikit_image==0.13.1
  • caffe2==0.8.1
  • skimage==0.0

How To

Download the pretrained models

	./download_models.sh

The script will download the LAP age and gender prediction models in caffe format.

Converting the original models

For example converting the LAP age model:

	cd models/lap
	python -m caffe2.python.caffe_translator \
		age.prototxt \
		dex_chalearn_iccv2015.caffemodel

This will create the needed init_net.pb and predict_net.pb needed for inference.

Similarly for the gender model:

	cd models/gender
	python -m caffe2.python.caffe_translator \
		gender.prototxt \
		gender.caffemodel

Run

	python run.py

Resources

If you are using this codebase or the provided train models please cite the authors:

@article{Rothe-IJCV-2016,
  author = {Rasmus Rothe and Radu Timofte and Luc Van Gool},
  title = {Deep expectation of real and apparent age from a single image without facial landmarks},
  journal = {International Journal of Computer Vision (IJCV)},
  year = {2016},
  month = {July},
}

@InProceedings{Rothe-ICCVW-2015,
  author = {Rasmus Rothe and Radu Timofte and Luc Van Gool},
  title = {DEX: Deep EXpectation of apparent age from a single image},
  booktitle = {IEEE International Conference on Computer Vision Workshops (ICCVW)},
  year = {2015},
  month = {December},
}

Papers:

Datasets:

Appa-real

IMDB - wiki 500:

Pre-trained models (caffe):

Caffe2 & Caffe2PyTorch conversion:

Misc:

About

Age and Gender estimation using Caffe2 pretrained LAP Challenge models

Topics

Resources

Stars

0 stars

Watchers

0 watching

Forks

Releases

Packages

Used by

Contributors

Languages