You can deploy this model and start earning money today!

a23f2003713a/
age-gender-resnet18

Views

No views yet

🧠 Finetuned ResNet18 — Age & Gender Classification

This repository contains a finetuned ResNet18 PyTorch model trained for predicting age group and gender from facial images.

📌 Model Overview

Base architecture: ResNet18 (ImageNet pretrained)
Framework: PyTorch
Task: Image Classification
Output: Age group & Gender
File: resnet18_finetuned_age_gender.pth
License: Apache 2.0

📥 How to Use This Model

1️⃣ Load ResNet18

python

import torch
from torchvision import models

model = models.resnet18()
state_dict = torch.hub.load_state_dict_from_url(
    "https://huggingface.co/a23f2003713a/age-gender-resnet18/resolve/main/resnet18_finetuned_age_gender.pth",
    map_location="cpu"
)
model.load_state_dict(state_dict)
model.eval()

## 📊 Applications
- Facial demographic analytics  
- Age estimation  
- Gender classification  
- Pre-training for face attribute recognition  
- Computer vision educational projects  

---

## 📜 Training Info
The model was finetuned on an age/gender image dataset using:
- Adam optimizer  
- Cross entropy loss  
- Image augmentations  
- ResNet18 backbone  

---

## 📂 Files in this Model Repo
- `resnet18_finetuned_age_gender.pth` — model weights
- `README.md` — description and usage

---

### ✨ Author
Model uploaded using Kaggle + HuggingFace Hub.