Age and Gender Prediction using Deep cnns and Transfer Learning




Download 299,02 Kb.
Pdf ko'rish
bet8/13
Sana22.12.2023
Hajmi299,02 Kb.
#127116
1   ...   5   6   7   8   9   10   11   12   13
Training & Testing. In order to save training time each set was separately forward 
passed via each model to get corresponding 9 Numpy ndarrays as extracted input fea-
ture vectors and saved. Since the faces were already aligned and cropped no further 
preprocessing was carried out and input dimensions are kept same as original RGB 
photos i.e., 200 x 200 x 3. For gender classification, the loss is binary cross-entropy 
function. Class weights were also taken into account while training to make up for slight 
class imbalance as there are roughly 48% female and 52% male in the both the training 
and validation set. For age estimation, being a regression task, the loss function was 
mean squared error. The optimizer used in both cases is the AMSGrad variant of Adam 
[15] with an initial learning rate of 0.001 which is halved in the ending phase of training 



for better convergence. The choice of optimizer was based on the experiments carried 
out while training our custom CNN architecture and theory [15]. 

Evaluation 
The performance of the age estimation algorithms is evaluated based on the closeness 
of the predicted value to the actual value. The metrics widely used for the age estimation 
as a regression task is the mean absolute error or MAE which captures the average 
magnitude of error in a set of predictions. MAE calculates the absolute error between 
actual age and predicted age as defined by the equation (1). 
𝑀𝐴𝐸 =
1
𝑛

|𝑦
𝑗
− 𝑦
𝑗
̂ |
𝑛
𝑗=1
(1)
Where n is the number of testing samples, 
𝑦
𝑗
denotes the ground truth age and 
𝑦
𝑗
̂ is the 
predicted age of the j-th sample. 
For classification tasks (age and gender), the evaluation metric used was accuracy 
which denotes the fraction of correctly classified samples over the total number of sam-
ples. 

Download 299,02 Kb.
1   ...   5   6   7   8   9   10   11   12   13




Download 299,02 Kb.
Pdf ko'rish

Bosh sahifa
Aloqalar

    Bosh sahifa



Age and Gender Prediction using Deep cnns and Transfer Learning

Download 299,02 Kb.
Pdf ko'rish