---
tags:
- yolov8
- object-detection
- computer-vision
- custom-dataset
- huggingface
- ultralytics
license: apache-2.0
datasets:
- custom
model-index:
- name: Celebrity Detection YOLOv8n
  results:
    - task:
        type: object-detection
      dataset:
        name: Custom Dataset (Friends)
        type: image
      metrics:
        - type: mAP
          value: 0.98
        - type: mAP50-95
          value: 0.91
        - type: precision
          value: 0.96
        - type: recall
          value: 0.98
---

# Celebrity Detection YOLOv8n

📌 **[English](#celebrity-detection-yolov8n) | [Español](#espanol)**

---

## 📄 Celebrity Detection YOLOv8n (English)
⬇ Scroll down for [Español](#espanol)

This repository contains a **YOLOv8n** model trained to detect the main characters from the TV show **Friends** in images and videos. The model was trained on a custom dataset labeled using Amazon Rekognition.

## 📌 Model Details

- **Model:** YOLOv8n (pretrained on COCO, fine-tuned on Friends dataset)
- **Classes:**
  - Matt LeBlanc (Joey)
  - David Schwimmer (Ross)
  - Matthew Perry (Chandler)
  - Courteney Cox (Monica)
  - Jennifer Aniston (Rachel)
- **Dataset:** Custom dataset generated from **YouTube shorts** of Friends episodes
- **Training Images:** ~13,000 (80% training, 20% validation)
- **Hardware Used:** Trained on GPU
- **Frameworks & Libraries:** `ultralytics`

## 🏆 Model Performance

| Metric            | Value |
|------------------|-------|
| **mAP@50**       | 0.98  |
| **mAP@50-95**    | 0.91  |
| **Precision**    | 0.96  |
| **Recall**       | 0.98  |

### Performance Curves:

**F1-Confidence Curve:**
![F1 Curve](./images/F1_curve.png)

**Recall-Confidence Curve:**
![Recall Curve](./images/R_curve.png)

**Precision-Confidence Curve:**
![Precision Curve](./images/P_curve.png)

**Confusion Matrix:**
![Confusion Matrix](./images/confusion_matrix_normalized.png)

## 📸 Example Detections

Correctly detected images:
![Detection Example](./images/image1.png)
![Detection Example](./images/image2.png)

Incorrect/misclassified example (blonde woman mistaken for Jennifer Aniston):
![Misclassification](./images/image3.png)

---

<a id="espanol"></a>
## 📄 Modelo de Detección de Celebridades YOLOv8n (Español)
⬆ Scroll up for [English](#celebrity-detection-yolov8n)

Este repositorio contiene un modelo **YOLOv8n** entrenado para detectar a los personajes principales de la serie **Friends** en imágenes y videos. El modelo fue entrenado en un conjunto de datos personalizado etiquetado con Amazon Rekognition.

## 🏆 Rendimiento del Modelo

| Métrica          | Valor |
|-----------------|-------|
| **mAP@50**     | 0.98  |
| **mAP@50-95**  | 0.91  |
| **Precisión**  | 0.96  |
| **Recall**     | 0.98  |

### Curvas de Rendimiento:

**Curva F1-Confianza:**
![Curva F1](./images/F1_curve.png)

**Curva Recall-Confianza:**
![Curva Recall](./images/R_curve.png)

**Curva Precisión-Confianza:**
![Curva Precisión](./images/P_curve.png)

**Matriz de Confusión:**
![Matriz de Confusión](./images/confusion_matrix_normalized.png)

## 📸 Ejemplos de Detección

Imágenes detectadas correctamente:
![Ejemplo de Detección](./images/image1.png)
![Ejemplo de Detección](./images/image2.png)

Ejemplo de detección errónea (mujer rubia identificada incorrectamente como Jennifer Aniston):
![Error de Clasificación](./images/image3.png)

---

¡Déjanos saber tus comentarios o mejoras! 🚀