detección de edad a partir de rostros

dima806

Clasificación de imagen

Este modelo devuelve el grupo de edad basado en una imagen facial. Para más detalles, consulte https://www.kaggle.com/code/dima806/age-group-image-detection-vit.

Como usar

from transformers import AutoModelForImageClassification, AutoProcessor
import torch
from PIL import Image

model = AutoModelForImageClassification.from_pretrained('dima806/faces_age_detection')
processor = AutoProcessor.from_pretrained('dima806/faces_age_detection')

img = Image.open('path_to_your_image.jpg')
inputs = processor(images=img, return_tensors='pt')

with torch.no_grad():
    logits = model(**inputs).logits

predicted_class_id = logits.argmax(-1).item()
print(model.config.id2label[predicted_class_id])

Funcionalidades

Clasificación de imágenes
Transformadores
PyTorch
Safetensors

Casos de uso

Clasificación de la edad en imágenes faciales
Análisis demográfico en imágenes