[FEATURE] Model (best.pt and last.pt) and model card upload for yolov8n celebrity detection model

README.md

@@ -1,3 +1,101 @@
+# Celebrity Detection YOLOv8n
+
+📌 **[English](#celebrity-detection-yolov8n) | [Español](#espanol)**
+
 ---
-license: apache-2.0
+
+## 📄 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! 🚀
+

images/.gitattributes

@@ -0,0 +1 @@
+*.png filter=lfs diff=lfs merge=lfs -text

models/.gitattributes

@@ -0,0 +1 @@
+*.pt filter=lfs diff=lfs merge=lfs -text

models/best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:63a048a7db16da856bdc1c51488ab82677859bb8412312040342fc9e7c631af3
+size 6232803

models/last.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:246234d6aed795fb068e135bb94b7dc1fd39c08f5af4475407a23c93679984ec
+size 6232803