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
Matriz de Confusión Normalizada
Matriz de Confusión Normalizada
Curvas de Evaluación
Curva PR (Precisión-Recall)
Curva PR
Curva F1
Curva F1
Curva de Precisión
Curva de Precisión
Curva R (Recall)
Curva R
Imágenes de Entrenamiento y Validación
Ejemplo de imágenes de entrenamiento
Train Batch 0
Train Batch 1
Train Batch 1980
Ejemplo de imágenes de validación
Validación Batch 0 - Labels
Validación Batch 1 - Predicciones
Validación Batch 2 - Predicciones
Instalación y Uso
Para utilizar este modelo, puedes cargarlo desde Hugging Face con: