BhoomiAg/age-gender-prediction-cnn

Age and Gender Prediction Model

A simple CNN model for predicting age (regression) and gender (classification) from facial images.

Usage

import torch
from model import SimpleCNN

model = SimpleCNN()
model.load_state_dict(torch.load('model.safetensors'))
model.eval()

Model Details

• Input: 224x224 RGB images
• Output: Age (continuous), Gender (0=female, 1=male)
• Parameters: 643,362
• Task: Multi-task learning (regression + classification)

Architecture

• Custom CNN backbone with 4 convolutional blocks
• Separate heads for age regression and gender classification
• Dropout for regularization

Training

This is a baseline model architecture. For best results, fine-tune on your specific dataset.