Downloads · 30 days
5
23% of all-time downloads
age-gender-prediction-cnn
BhoomiAg/age-gender-prediction-cnn
age-gender-prediction-cnn is a machine learning model from BhoomiAg. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
A simple CNN model for predicting age (regression) and gender (classification) from facial images.
All-time downloads
22
Public
Repo size
2.6 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors2.6 MB · 100%
At a glance
- Model type
- SimpleCNN
- Access
- Public
- Created
- Nov 14, 2025
- Updated
- Nov 14, 2025
- SHA
- bb10e435
Model card
From the Hugging Face model README
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.
- Type
- SimpleCNN
- Created
- Nov 14, 2025
- Updated
- Nov 14, 2025