iabd2425-reto2-grupo5/x-ray-aws-objects

Modelo de Detección de Caras de Actores de Harry Potter

Descripción

Este modelo de detección de caras está basado en YOLO11 y ha sido entrenado para reconocer los rostros de actores de la saga de películas de Harry Potter. Utiliza un conjunto de datos curado con imágenes de los personajes principales y está optimizado para la detección en tiempo real.

Uso

Este modelo puede utilizarse en aplicaciones de reconocimiento facial, análisis de escenas y sistemas de recomendación basados en contenido visual.

Arquitectura del Modelo

El modelo está basado en YOLO11, una arquitectura de última generación para detección de objetos, optimizada para velocidad y precisión.

Datos

El modelo fue entrenado con un conjunto de datos que contiene imágenes de múltiples personajes de la saga de Harry Potter, anotadas con bounding boxes generadas mayormente por aws rekognition para aumentar la cantidad y calidad del dataset.

Evaluación

El modelo ha sido evaluado utilizando métricas estándar como precisión (Precision), recall, F1-score y curva PR. A continuación, se presentan algunas visualizaciones clave de los resultados:

Matriz de Confusión

Matriz de Confusión Normalizada

Curvas de Evaluación

• Curva PR (Precisión-Recall)
• Curva F1
• Curva de Precisión
• Curva R (Recall)

Imágenes de Entrenamiento y Validación

• Ejemplo de imágenes de entrenamiento

• Ejemplo de imágenes de validación

Instalación y Uso

Para utilizar este modelo, puedes cargarlo desde Hugging Face con:

from ultralytics import YOLO

model = YOLO('ruta/al/modelo.pt')
results = model('ruta/a/imagen.jpg')
results.show()

Contribuciones

Si deseas contribuir al desarrollo o mejorar el modelo, por favor abre un issue o envía un pull request en el repositorio correspondiente.