genderage2
ivensamdh
Clasificación de imagen
Este modelo es una versión afinada de google/vit-base-patch16-224-in21k en el conjunto de datos imagefolder. Logra los siguientes resultados en el conjunto de evaluación: Pérdida: 0.2771.
Como usar
# Uso del modelo genderage2
from transformers import AutoModel, AutoImageProcessor
image_processor = AutoImageProcessor.from_pretrained('ivensamdh/genderage2')
model = AutoModel.from_pretrained('ivensamdh/genderage2')
# Cargar y procesar una imagen
image = ... # cargar la imagen de alguna fuente
inputs = image_processor(images=image, return_tensors='pt')
outputs = model(**inputs)
Funcionalidades
- Clasificación de imágenes
- Transformers
- PyTorch
- Entrenado con imagenes de un folder específico
Casos de uso
- Clasificación de género y edad de las imágenes