Skip to content

cunjian/multitask_CNN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Introduction

A project that applies CNN to simultaneously predict facial attributes, including gender and age. This can be extended to more attributes.

This multitask CNN has been trained on scratch from Celebrity dataset released by CUHK. After 500000 iterations, it reaches 84.21% for age classification and 95.47% for gender classification.

Evaluation Results

Male: M; Female: F; Young: Y; Old: O

Along with the attribute symbols, the network also outputs the probability associated with each attribute.

alt text

alt text

Usage

Install caffe and dlib, run the following command

python predict_vanilla_fd_attribute.py ../train_multitask/model/_iter_500000.caffemodel family3.jpg

About

CAFFE MULTITASK CNN

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages