You can deploy this model and start earning money today!

abhilash88/
age-gender-prediction

Name: age-gender-prediction
Availability: InStock

gender-classification

Views

No views yet

🏆 ViT Age-Gender Prediction: Vision Transformer for Facial Analysis

A state-of-the-art Vision Transformer model for simultaneous age estimation and gender classification, achieving 94.3% gender accuracy and 4.5 years age MAE on the UTKFace dataset.

🚀 One-Liner Usage

python

from model import predict_age_gender

result = predict_age_gender("your_image.jpg")
print(f"Age: {result['age']}, Gender: {result['gender']}")

That's it! One line to get age and gender predictions.

🆕 October 2025 Update - Discussion #5 Fixed

✅ Issue Resolved: Model now includes helper functions that return proper age and gender values (not LABEL_0/LABEL_1)

Recommended usage:

python

from model import predict_age_gender

result = predict_age_gender("image.jpg")
print(f"Age: {result['age']}, Gender: {result['gender']}")
print(f"Confidence: {result['gender_confidence']:.1%}")

Simple one-liner version:

python

from model import simple_predict

print(simple_predict("image.jpg"))
# Output: "25 years, Female (87.3% confidence)"

Important: These helper functions work correctly. The standard pipeline() approach returns LABEL_0/LABEL_1 and should not be used.

📱 Complete Examples

Basic Usage

python

from model import predict_age_gender

# Predict from file
result = predict_age_gender("your_image.jpg")
print(f"Age: {result['age']} years")
print(f"Gender: {result['gender']}")
print(f"Confidence: {result['gender_confidence']:.1%}")

# Predict from URL
result = predict_age_gender("https://example.com/face_image.jpg")
print(f"Prediction: {result['age']} years, {result['gender']}")

# Works with PIL Image too
from PIL import Image
img = Image.open("image.jpg")
result = predict_age_gender(img)
print(f"Result: {result['age']} years, {result['gender']}")

Simple Helper Functions

python

from model import predict_age_gender, simple_predict

# Method 1: Detailed result
result = predict_age_gender("your_image.jpg")
print(f"Age: {result['age']}, Gender: {result['gender']}")
print(f"Confidence: {result['confidence']:.1%}")

# Method 2: Simple string output
prediction = simple_predict("your_image.jpg")
print(prediction)  # "25 years, Female (87% confidence)"

Google Colab

python

# Install requirements
!pip install transformers torch pillow

from model import predict_age_gender
import matplotlib.pyplot as plt
from PIL import Image

# Upload image in Colab
from google.colab import files
uploaded = files.upload()
filename = list(uploaded.keys())[0]

# Predict
result = predict_age_gender(filename)

# Display
img = Image.open(filename)
plt.figure(figsize=(8, 6))
plt.imshow(img)
plt.title(f"Prediction: {result['age']} years, {result['gender']} ({result['gender_confidence']:.1%})")
plt.axis('off')
plt.show()

print(f"Age: {result['age']} years")
print(f"Gender: {result['gender']}")
print(f"Confidence: {result['gender_confidence']:.1%}")

Batch Processing

python

from model import predict_age_gender

# Process multiple images
images = ["image1.jpg", "image2.jpg", "image3.jpg"]
results = []

for image in images:
    result = predict_age_gender(image)
    results.append({
        'image': image,
        'age': result['age'],
        'gender': result['gender'],
        'confidence': result['gender_confidence']
    })

for result in results:
    print(f"{result['image']}: {result['age']} years, {result['gender']} ({result['confidence']:.1%})")

Real-time Webcam

python

import cv2
from model import predict_age_gender
from PIL import Image

cap = cv2.VideoCapture(0)

while True:
    ret, frame = cap.read()
    if ret:
        # Convert frame to PIL Image
        rgb_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
        pil_image = Image.fromarray(rgb_frame)
        
        # Predict
        result = predict_age_gender(pil_image)
        
        # Display prediction
        text = f"Age: {result['age']}, Gender: {result['gender']}"
        cv2.putText(frame, text, (10, 30), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 0), 2)
        cv2.imshow('Age-Gender Detection', frame)
        
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

cap.release()
cv2.destroyAllWindows()

URL Images

python

from model import predict_age_gender

# Direct URL prediction
image_url = "https://images.unsplash.com/photo-1507003211169-0a1dd7228f2d?w=300"
result = predict_age_gender(image_url)

print(f"Age: {result['age']} years")
print(f"Gender: {result['gender']}")
print(f"Confidence: {result['gender_confidence']:.1%}")

📊 Output Format

The helper function returns a dictionary with the prediction:

python

{
    "age": 25,
    "gender": "Female", 
    "gender_confidence": 0.873,
    "gender_probability_male": 0.127,
    "gender_probability_female": 0.873,
    "label": "25 years, Female",
    "score": 0.873
}

Access the values:

result['age'] - Predicted age (integer, 0-100)
result['gender'] - Predicted gender ("Male" or "Female")
result['gender_confidence'] - Confidence score (0-1)
result['gender_probability_male'] - Male probability (0-1)
result['gender_probability_female'] - Female probability (0-1)
result['label'] - Formatted string summary

🎯 Model Performance

Metric	Performance	Dataset
Gender Accuracy	94.3%	UTKFace
Age MAE	4.5 years	UTKFace
Architecture	ViT-Base + Dual Head	768→256→64→1
Parameters	86.8M	Optimized
Inference Speed	~50ms/image	CPU

Performance by Age Group

Adults (21-60 years): 94.3% gender accuracy, 4.5 years age MAE ✅ Excellent
Young Adults (16-30 years): 92.1% gender accuracy ✅ Very Good
Teenagers (13-20 years): 89.7% gender accuracy ✅ Good
Children (5-12 years): 78.4% gender accuracy ⚠️ Limited
Seniors (60+ years): 87.2% gender accuracy ✅ Good

⚠️ Usage Guidelines

✅ Optimal Performance

Best for: Adults 16-60 years old
Image quality: Clear, well-lit, front-facing faces
Use cases: Demographic analysis, content filtering, marketing research

❌ Known Limitations

Children (0-12): Reduced accuracy due to limited training data
Very elderly (70+): Higher prediction variance
Poor conditions: Low light, extreme angles, heavy occlusion

🎯 Tips for Best Results

Use clear, well-lit images
Ensure faces are clearly visible and front-facing
Consider confidence scores for critical applications
Validate results for your specific use case

🛠️ Installation

bash

# Minimal installation
pip install transformers torch pillow

# Full installation with optional dependencies  
pip install transformers torch torchvision pillow opencv-python matplotlib

# For development
pip install transformers torch pillow pytest black flake8

📈 Use Cases & Examples

Content Moderation

python

from model import predict_age_gender

def moderate_content(image_path):
    result = predict_age_gender(image_path)
    age = result['age']
    
    if age < 18:
        return f"Minor detected ({age} years) - content flagged for review"
    return f"Adult content approved: {age} years, {result['gender']}"

status = moderate_content("user_upload.jpg")
print(status)

Marketing Analytics

python

from model import predict_age_gender
from glob import glob

def analyze_audience(image_folder):
    demographics = {"male": 0, "female": 0, "total_age": 0, "count": 0}
    
    for image_path in glob(f"{image_folder}/*.jpg"):
        result = predict_age_gender(image_path)
        demographics[result['gender'].lower()] += 1
        demographics['total_age'] += result['age']
        demographics['count'] += 1
    
    demographics['avg_age'] = demographics['total_age'] / demographics['count']
    demographics['male_percent'] = demographics['male'] / demographics['count'] * 100
    demographics['female_percent'] = demographics['female'] / demographics['count'] * 100
    
    return demographics

stats = analyze_audience("customer_photos/")
print(f"Average age: {stats['avg_age']:.1f}")
print(f"Gender split: {stats['male_percent']:.1f}% Male, {stats['female_percent']:.1f}% Female")

Age Verification

python

from model import predict_age_gender

def verify_age(image_path, min_age=18):
    result = predict_age_gender(image_path)
    age = result['age']
    confidence = result['gender_confidence']
    
    if confidence < 0.7:  # Low confidence
        return "Please provide a clearer image"
    
    if age >= min_age:
        return f"Verified: {age} years old (meets {min_age}+ requirement)"
    else:
        return f"Age verification failed: {age} years old"

verification = verify_age("id_photo.jpg", min_age=21)
print(verification)

🔧 Technical Details

Base Model: google/vit-base-patch16-224 (Vision Transformer)
Input Resolution: 224×224 RGB images
Architecture: Dual-head design with age regression and gender classification
Training Dataset: UTKFace (23,687 images)
Training: 15 epochs, AdamW optimizer, 2e-5 learning rate

🌟 Key Features

✅ True one-line usage with transformers pipeline
✅ High accuracy (94.3% gender, 4.5 years age MAE)
✅ Multiple input types (file paths, URLs, PIL Images, NumPy arrays)
✅ Batch processing support
✅ Real-time capable (~50ms inference)
✅ Google Colab ready
✅ Production tested

🚀 Quick Start Examples

Absolute Minimal Usage

python

from model import predict_age_gender
result = predict_age_gender("image.jpg")
print(f"Age: {result['age']}, Gender: {result['gender']}")

With Helper Function

python

from model import simple_predict
print(simple_predict("image.jpg"))  # "25 years, Female (87% confidence)"

Error Handling

python

from model import predict_age_gender

def safe_predict(image_path):
    try:
        result = predict_age_gender(image_path)
        return f"Age: {result['age']}, Gender: {result['gender']}"
    except Exception as e:
        return f"Prediction failed: {e}"

prediction = safe_predict("any_image.jpg")
print(prediction)

🔧 Troubleshooting

Issue: Getting `LABEL_0`/`LABEL_1` instead of age/gender

Solution: Use the helper functions instead of pipeline:

python

# ✅ CORRECT METHOD - Use helper function
from model import predict_age_gender

result = predict_age_gender("image.jpg")
print(f"Age: {result['age']}, Gender: {result['gender']}")
# Output: Age: 25, Gender: Female

python

# ❌ WRONG METHOD - Don't use standard pipeline
from transformers import pipeline
classifier = pipeline("image-classification", ...)  # Returns LABEL_0/LABEL_1

The standard pipeline() approach doesn't work properly with custom models. Always use the predict_age_gender() helper function.

Issue: Warning "Some weights not initialized"

This warning is expected and safe to ignore:

Some weights of ViTForImageClassification were not initialized...

The model uses custom age and gender heads instead of standard classification, which causes this informational warning. The model works correctly.

Issue: Low confidence predictions

For optimal results:

✅ Use clear, well-lit images
✅ Ensure face is front-facing and visible
✅ Avoid heavy occlusion or extreme angles
⚠️ Predictions with confidence < 0.7 may need manual review

📝 Citation

bibtex

@misc{age-gender-prediction-2025,
  title={Age-Gender-Prediction: Vision Transformer for Facial Analysis},
  author={Abhilash Sahoo},
  year={2025},
  publisher={Hugging Face},
  url={https://huggingface.co/abhilash88/age-gender-prediction},
  note={One-liner pipeline with 94.3\% gender accuracy}
}

📄 License

Licensed under Apache 2.0. Commercial use permitted with attribution.

🎉 Ready to use! Just one line of code to get accurate age and gender predictions from any facial image! 🚀

Try it now:

python

from model import predict_age_gender

result = predict_age_gender("your_image.jpg")
print(f"Age: {result['age']}, Gender: {result['gender']}")
print(f"Confidence: {result['gender_confidence']:.1%}")

Simple one-liner:

python

from model import simple_predict
print(simple_predict("your_image.jpg"))
# Output: "25 years, Female (87.3% confidence)"

abhilash88/age-gender-prediction

🏆 ViT Age-Gender Prediction: Vision Transformer for Facial Analysis

🚀 One-Liner Usage

🆕 October 2025 Update - Discussion #5 Fixed

📱 Complete Examples

Basic Usage

Simple Helper Functions

Google Colab

Batch Processing

Real-time Webcam

URL Images

📊 Output Format

🎯 Model Performance

Performance by Age Group

⚠️ Usage Guidelines

✅ Optimal Performance

❌ Known Limitations

🎯 Tips for Best Results

🛠️ Installation

📈 Use Cases & Examples

Content Moderation

Marketing Analytics

Age Verification

🔧 Technical Details

🌟 Key Features

🚀 Quick Start Examples

Absolute Minimal Usage

With Helper Function

Error Handling

🔧 Troubleshooting

Issue: Getting LABEL_0/LABEL_1 instead of age/gender

Issue: Warning "Some weights not initialized"

Issue: Low confidence predictions

📝 Citation

📄 License

abhilash88/
age-gender-prediction

Issue: Getting `LABEL_0`/`LABEL_1` instead of age/gender