Skip to content

Age and Gender estimation using Caffe2 pretrained LAP Challenge models

Notifications You must be signed in to change notification settings

josemarcosrf/Age-Gender-Estimation-example

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 
 
 
 
 

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

Watchers

Forks

Releases

No releases published

Packages

No packages published