Git Product home page Git Product logo

age-estimation-by-cnn's Introduction

Age Estimation by CNN Based Regression Model

  • Completed by: Hamdi Alperen Çetin & Emre Doğan

  • We propose a Convolutional Neural Network Model to succesfully estimate the age of a person given her/his cropped face image. Different from the classical CNN models, our model ends up with a regression layer, not a classifier one. So, backpropagation process is done based on the regression output.

  • For a more detailed technical report, check here.

Dataset:

  • We trained our model with a downsampled version of UTKFace Dataset.
  • Due to its large size, we cannot share original dataset(all training + validation + test data). But you can find some samples of our dataset from here. Notice that the first 3 letters of any image corresponds to its output layer (age of the person in the image).
  • Not to spend time on reading data on each execution, we converted our training, validation and test data into .npy format by read_data.py.

Model:

alt text

  • Our architecture can be seen in the figure above. It consists of several consecutive convolution layers. Another important point regarding the model is that instead of a classifier approach, we used a regression based model so that backpropagation flow starts from some continuous age value.

To see a more detailed tensorboard graph regarding our model, click here.

Results

  • To decide on hyperparameters, we tried many different scenarios. Training and validation losses (Mean Average Error) for each scenario can be found here.

  • The best results are taken when the hyperparameters are,

Hyperparameter Choosen Value
Loss Function Mean Sqaure Error
Learning Rate 0.0001
Dropout Keep Probability 0.6
L2 Reg. Constant 0.0001
Batch Size 200
  • The corresponding results in our best model is given below,
Loss Type Mean Average Error
Validation Loss 6.486
Test Loss 6.419

age-estimation-by-cnn's People

Contributors

dependabot[bot] avatar emredogan7 avatar hacetin avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar

age-estimation-by-cnn's Issues

How to prepare data for training ?

Hello,
I was going through your code to convert the dataset into .npy file format and came across the code lines
path_train = "./../data/train/"
path_test = "./../data/test/"
path_validation = "./../data/validation/"

Do I have to split the UTKFace dataset into Train Test and Validation sets by my self and put those into respective folders before running the read_data.py?

Also, I would like to know that where was your paper published as I want to use it as a reference. I would be glad if you could share your publication details.
Thanks in Advance. Pls Reply ASAP.

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.