Merge de yolov8-model a develop con historias no relacionadas
Browse files- .gitattributes +22 -0
- .gitignore +51 -0
- README.md +361 -52
- predict.py +36 -0
- processed-video/ny-traffic-processed.mp4 +3 -0
- processed-video/ny-walking-processed.mp4 +3 -0
- raw-video/ny-traffic.mp4 +3 -0
- raw-video/ny-walking.mp4 +3 -0
- requirements.txt +1 -0
- runs/val/best.pt +3 -0
- runs/val/last.pt +3 -0
- runs/val/val_coco8/F1_curve.png +0 -0
- runs/val/val_coco8/PR_curve.png +0 -0
- runs/val/val_coco8/P_curve.png +0 -0
- runs/val/val_coco8/R_curve.png +0 -0
- runs/val/val_coco8/confusion_matrix.png +0 -0
- runs/val/val_coco8/confusion_matrix_normalized.png +0 -0
- runs/val/val_coco8/val_batch0_labels.jpg +0 -0
- runs/val/val_coco8/val_batch0_pred.jpg +0 -0
- train_yolov8n.py +24 -0
- train_yolov8s.py +20 -0
- validate.py +15 -0
.gitattributes
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
*.7z filter=lfs diff=lfs merge=lfs -text
|
| 2 |
*.arrow filter=lfs diff=lfs merge=lfs -text
|
| 3 |
*.bin filter=lfs diff=lfs merge=lfs -text
|
|
@@ -33,3 +34,24 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
<<<<<<< HEAD
|
| 2 |
*.7z filter=lfs diff=lfs merge=lfs -text
|
| 3 |
*.arrow filter=lfs diff=lfs merge=lfs -text
|
| 4 |
*.bin filter=lfs diff=lfs merge=lfs -text
|
|
|
|
| 34 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 36 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 37 |
+
=======
|
| 38 |
+
# Tratar datasets y runs como binarios para evitar conversiones de texto
|
| 39 |
+
datasets/combined/** filter=lfs diff=lfs merge=lfs -text
|
| 40 |
+
runs/detect/** filter=lfs diff=lfs merge=lfs -text
|
| 41 |
+
# Asegurar formato de texto en archivos clave
|
| 42 |
+
*.yaml text eol=lf
|
| 43 |
+
*.py text eol=lf
|
| 44 |
+
*.md text eol=lf
|
| 45 |
+
# Forzar archivos de configuración y modelos a usar Git LFS
|
| 46 |
+
*.pt filter=lfs diff=lfs merge=lfs -text
|
| 47 |
+
*.onnx filter=lfs diff=lfs merge=lfs -text
|
| 48 |
+
*.tflite filter=lfs diff=lfs merge=lfs -text
|
| 49 |
+
*.engine filter=lfs diff=lfs merge=lfs -text
|
| 50 |
+
*.ckpt filter=lfs diff=lfs merge=lfs -text
|
| 51 |
+
# Archivos de vídeo y datos grandes
|
| 52 |
+
*.mp4 filter=lfs diff=lfs merge=lfs -text
|
| 53 |
+
*.csv filter=lfs diff=lfs merge=lfs -text
|
| 54 |
+
*.json filter=lfs diff=lfs merge=lfs -text
|
| 55 |
+
processed-video/*.mp4 filter=lfs diff=lfs merge=lfs -text
|
| 56 |
+
raw-video/*.mp4 filter=lfs diff=lfs merge=lfs -text
|
| 57 |
+
>>>>>>> yolov8-model
|
.gitignore
ADDED
|
@@ -0,0 +1,51 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Ignorar la carpeta de entornos virtuales
|
| 2 |
+
.venv/
|
| 3 |
+
venv/
|
| 4 |
+
|
| 5 |
+
# Archivos de caché de Python
|
| 6 |
+
__pycache__/
|
| 7 |
+
*.py[cod]
|
| 8 |
+
*$py.class
|
| 9 |
+
|
| 10 |
+
# Archivos temporales del sistema
|
| 11 |
+
.DS_Store
|
| 12 |
+
Thumbs.db
|
| 13 |
+
|
| 14 |
+
# Archivos de configuración de usuario
|
| 15 |
+
*.log
|
| 16 |
+
*.tmp
|
| 17 |
+
*.swp
|
| 18 |
+
*.swo
|
| 19 |
+
|
| 20 |
+
# Claves API o credenciales privadas (⚠️ Asegúrate de que este archivo contiene datos sensibles)
|
| 21 |
+
download-roboflow.py
|
| 22 |
+
|
| 23 |
+
# PERMITIR subir `runs/val/`
|
| 24 |
+
!runs/val/
|
| 25 |
+
|
| 26 |
+
# Ignorar archivos de ejecución y checkpoints de modelos
|
| 27 |
+
runs/detect
|
| 28 |
+
|
| 29 |
+
weights/
|
| 30 |
+
*.pt
|
| 31 |
+
*.onnx
|
| 32 |
+
*.tflite
|
| 33 |
+
*.engine
|
| 34 |
+
*.ckpt
|
| 35 |
+
|
| 36 |
+
# EXCEPCIÓN: No ignorar los .pt dentro de runs/val
|
| 37 |
+
!runs/val/*.pt
|
| 38 |
+
|
| 39 |
+
# Ignorar dataset original de COCO para evitar archivos innecesarios
|
| 40 |
+
datasets/
|
| 41 |
+
download-coco.py
|
| 42 |
+
datasets-download/
|
| 43 |
+
|
| 44 |
+
# Ignorar archivos con la terminación .Zone.Identifier (propio de Windows)
|
| 45 |
+
*.Zone.Identifier
|
| 46 |
+
|
| 47 |
+
# PERMITIR subir `datasets/combined/`
|
| 48 |
+
# (Eliminar esta línea si accidentalmente se ignora)
|
| 49 |
+
!datasets/combined/
|
| 50 |
+
!datasets/combined/images/
|
| 51 |
+
!datasets/combined/labels/
|
README.md
CHANGED
|
@@ -1,52 +1,361 @@
|
|
| 1 |
-
|
| 2 |
-
|
| 3 |
-
|
| 4 |
-
|
| 5 |
-
|
| 6 |
-
-
|
| 7 |
-
-
|
| 8 |
-
-
|
| 9 |
-
-
|
| 10 |
-
-
|
| 11 |
-
|
| 12 |
-
|
| 13 |
-
|
| 14 |
-
|
| 15 |
-
|
| 16 |
-
|
| 17 |
-
|
| 18 |
-
|
| 19 |
-
|
| 20 |
-
- **
|
| 21 |
-
- **
|
| 22 |
-
- **
|
| 23 |
-
- **
|
| 24 |
-
- **
|
| 25 |
-
|
| 26 |
-
|
| 27 |
-
|
| 28 |
-
|
| 29 |
-
|
| 30 |
-
|
| 31 |
-
|
| 32 |
-
|
| 33 |
-
|
| 34 |
-
|
| 35 |
-
|
| 36 |
-
|
| 37 |
-
|
| 38 |
-
|
| 39 |
-
|
| 40 |
-
|
| 41 |
-
|
| 42 |
-
|
| 43 |
-
|
| 44 |
-
|
| 45 |
-
|
| 46 |
-
|
| 47 |
-
|
| 48 |
-
|
| 49 |
-
|
| 50 |
-
|
| 51 |
-
|
| 52 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
<<<<<<< HEAD
|
| 2 |
+
---
|
| 3 |
+
license: mit
|
| 4 |
+
---
|
| 5 |
+
tags:
|
| 6 |
+
- yolov8
|
| 7 |
+
- object-detection
|
| 8 |
+
- deep-learning
|
| 9 |
+
- computer-vision
|
| 10 |
+
- pretrained
|
| 11 |
+
---
|
| 12 |
+
|
| 13 |
+
# 📦 YOLOv8 - Modelo de Detección de Objetos
|
| 14 |
+
|
| 15 |
+
Este modelo está basado en **YOLOv8**, una de las arquitecturas más avanzadas para la detección en tiempo real de objetos en imágenes y vídeos. Está entrenado en un dataset de múltiples clases para reconocer una variedad de objetos con alta precisión.
|
| 16 |
+
|
| 17 |
+
---
|
| 18 |
+
|
| 19 |
+
## 🛠️ **Detalles Técnicos**
|
| 20 |
+
- **Arquitectura**: YOLOv8 (You Only Look Once)
|
| 21 |
+
- **Dataset**: Conjunto de datos etiquetado con objetos de diversas categorías.
|
| 22 |
+
- **Épocas de entrenamiento**: 600
|
| 23 |
+
- **Tamaño de imagen**: 640x640 px
|
| 24 |
+
- **Precisión final (mAP@50)**: 85% en validación
|
| 25 |
+
- **FPS en GPU**: ~45 en inferencia
|
| 26 |
+
|
| 27 |
+
---
|
| 28 |
+
|
| 29 |
+
## 🚀 **Uso del Modelo**
|
| 30 |
+
Puedes cargar el modelo y realizar predicciones con la librería `ultralytics`.
|
| 31 |
+
|
| 32 |
+
### **📸 Para Imágenes**
|
| 33 |
+
```python
|
| 34 |
+
import cv2
|
| 35 |
+
from ultralytics import YOLO
|
| 36 |
+
|
| 37 |
+
model = YOLO("izaskun/yolov8-object-detection")
|
| 38 |
+
cap = cv2.VideoCapture("video.mp4")
|
| 39 |
+
|
| 40 |
+
while cap.isOpened():
|
| 41 |
+
ret, frame = cap.read()
|
| 42 |
+
if not ret:
|
| 43 |
+
break
|
| 44 |
+
|
| 45 |
+
results = model.predict(frame)
|
| 46 |
+
annotated_frame = results[0].plot()
|
| 47 |
+
cv2.imshow("YOLOv8 - Detección en Vídeo", annotated_frame)
|
| 48 |
+
|
| 49 |
+
if cv2.waitKey(1) == ord('q'):
|
| 50 |
+
break
|
| 51 |
+
|
| 52 |
+
cap.release()
|
| 53 |
+
cv2.destroyAllWindows()
|
| 54 |
+
=======
|
| 55 |
+
---
|
| 56 |
+
license: mit
|
| 57 |
+
base_model:
|
| 58 |
+
- Ultralytics/YOLOv8
|
| 59 |
+
tags:
|
| 60 |
+
- yolov8
|
| 61 |
+
- object-detection
|
| 62 |
+
- deep-learning
|
| 63 |
+
- computer-vision
|
| 64 |
+
- pretrained
|
| 65 |
+
---
|
| 66 |
+
|
| 67 |
+
# 📦 YOLOv8s - Modelo de Detección de Objetos
|
| 68 |
+
|
| 69 |
+
Este modelo está basado en **YOLOv8s**, entrenado específicamente para la detección de objetos en entornos urbanos y de tráfico. Se han combinado múltiples datasets para mejorar la detección de matrículas y objetos en escenarios urbanos complejos.
|
| 70 |
+
|
| 71 |
+
---
|
| 72 |
+
|
| 73 |
+
### 📂 Arquitectura del Proyecto
|
| 74 |
+
El modelo forma parte de un pipeline más amplio donde los videos son procesados en AWS. La arquitectura general es la siguiente:
|
| 75 |
+
|
| 76 |
+

|
| 77 |
+
|
| 78 |
+
1. Los videos son enviados a un **bucket S3** desde una API.
|
| 79 |
+
2. Un **AWS Lambda** enciende una instancia **EC2** que contiene el modelo YOLOv8s.
|
| 80 |
+
3. La EC2 procesa el video y envía los resultados a **DynamoDB**.
|
| 81 |
+
4. Los resultados finales se almacenan en **S3** en formato JSON y logs en formato .log.
|
| 82 |
+
5. DynamoDB indexa la información con claves secundarias globales (GSI).
|
| 83 |
+
6. Cuando el proceso finaliza, una **segunda Lambda** apaga la instancia EC2.
|
| 84 |
+
|
| 85 |
+
---
|
| 86 |
+
|
| 87 |
+
### 📊 Datasets Utilizados
|
| 88 |
+
Para entrenar el modelo, se ha utilizado el dataset de **COCO8**, pero también es posible añadir otros datasets como **License Plate Recognition** o **Shahbagh Traffic Dataset**:
|
| 89 |
+
|
| 90 |
+
- **COCO8** - versión reducida de COCO para pruebas rápidas.
|
| 91 |
+
- **License Plate Recognition** - para mejorar la detección de matrículas.
|
| 92 |
+
- **Shahbagh Traffic Dataset** - dataset específico para escenas de tráfico.
|
| 93 |
+
|
| 94 |
+
---
|
| 95 |
+
|
| 96 |
+
### ⚙ **Configuración del Entorno**
|
| 97 |
+
Para garantizar un entrenamiento sin problemas, es importante configurar correctamente el entorno. Se recomienda usar un entorno virtual de Python y asegurarse de que todas las dependencias necesarias estén instaladas.
|
| 98 |
+
|
| 99 |
+
#### **1️⃣ Crear y activar un entorno virtual (opcional pero recomendado)**
|
| 100 |
+
```bash
|
| 101 |
+
# Crear el entorno virtual
|
| 102 |
+
python -m venv .venv
|
| 103 |
+
|
| 104 |
+
# Activar el entorno virtual
|
| 105 |
+
# En Linux/macOS
|
| 106 |
+
source .venv/bin/activate
|
| 107 |
+
|
| 108 |
+
# En Windows (cmd o PowerShell)
|
| 109 |
+
.venv\Scripts\activate
|
| 110 |
+
```
|
| 111 |
+
|
| 112 |
+
#### **2️⃣ Instalar las dependencias necesarias**
|
| 113 |
+
```bash
|
| 114 |
+
pip install -r requirements.txt
|
| 115 |
+
```
|
| 116 |
+
|
| 117 |
+
o también
|
| 118 |
+
|
| 119 |
+
```bash
|
| 120 |
+
pip install ultralytics roboflow
|
| 121 |
+
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu # Cambiar a cu121 si se usa GPU con CUDA 12.1
|
| 122 |
+
```
|
| 123 |
+
|
| 124 |
+
---
|
| 125 |
+
|
| 126 |
+
#### **3️⃣ Descargar los datasets**
|
| 127 |
+
Si utilizas el **dataset de COCO8**, puedes descargarlo con el siguiente código:
|
| 128 |
+
|
| 129 |
+
```python
|
| 130 |
+
from ultralytics.utils.downloads import download
|
| 131 |
+
|
| 132 |
+
# Descargar el dataset COCO8 en formato YOLO
|
| 133 |
+
download('https://ultralytics.com/assets/coco8.zip', dir='datasets')
|
| 134 |
+
```
|
| 135 |
+
|
| 136 |
+
Si utilizas los **datasets de Roboflow**, puedes descargarlos con el siguiente código:
|
| 137 |
+
|
| 138 |
+
```python
|
| 139 |
+
from roboflow import Roboflow
|
| 140 |
+
|
| 141 |
+
# Configurar la API Key
|
| 142 |
+
rf = Roboflow(api_key="TU_API_KEY")
|
| 143 |
+
|
| 144 |
+
# Cargar el dataset desde Roboflow Universe
|
| 145 |
+
project = rf.workspace("shovonthesis").project("shahbagh-g7vmy")
|
| 146 |
+
|
| 147 |
+
# Seleccionar la versión 4 del dataset (según la URL)
|
| 148 |
+
version = project.version(4)
|
| 149 |
+
|
| 150 |
+
# Descargar el dataset en formato YOLOv8
|
| 151 |
+
dataset_path = version.download("yolov8")
|
| 152 |
+
|
| 153 |
+
print(f"✅ Dataset descargado en: {dataset_path}")
|
| 154 |
+
```
|
| 155 |
+
|
| 156 |
+
### 🔹 **Observaciones**
|
| 157 |
+
Para descargar los datasets de Roboflow mediante código es necesario la Private API Key que se encuentra en **Settings > APi Keys** de tu cuenta de Roboflow.
|
| 158 |
+
|
| 159 |
+
# `rf.workspace("license-project")`
|
| 160 |
+
- `rf` es un objeto de la clase **Roboflow** que hemos inicializado con nuestra **API Key**.
|
| 161 |
+
- `.workspace("shovonthesis")` selecciona el espacio de trabajo llamado `"shovonthesis"`, que es donde está almacenado el dataset dentro de **Roboflow**.
|
| 162 |
+
|
| 163 |
+
# `.project("license-plate-detection-project")`
|
| 164 |
+
- Dentro del espacio de trabajo `"shovonthesis"`, buscamos el dataset con el identificador `"shahbagh-g7vmy"`.
|
| 165 |
+
- `project` ahora representa este dataset específico y nos permitirá **acceder a sus versiones, descargarlo o gestionarlo**.
|
| 166 |
+
- Una vez esto este configurado solo nos quedará seleccionar la versión que queremos del dataset y ejecutarlo.
|
| 167 |
+
|
| 168 |
+
---
|
| 169 |
+
|
| 170 |
+
### ⚙ **Configuración del Entrenamiento**
|
| 171 |
+
El modelo fue entrenado utilizando **YOLOv8s** con los siguientes parámetros:
|
| 172 |
+
|
| 173 |
+
```python
|
| 174 |
+
from ultralytics import YOLO
|
| 175 |
+
|
| 176 |
+
# Cargar el modelo YOLOv8s preentrenado
|
| 177 |
+
model = YOLO("yolov8s.pt")
|
| 178 |
+
|
| 179 |
+
# Entrenar el modelo
|
| 180 |
+
model.train(
|
| 181 |
+
data="/home/USER/yolo/yolov8-object-detection/datasets/combined/data.yaml",
|
| 182 |
+
epochs=150,
|
| 183 |
+
batch=8,
|
| 184 |
+
imgsz=640,
|
| 185 |
+
device='cpu', # Cambiar a 'cuda' si hay GPU disponible
|
| 186 |
+
project="/home/USER/yolo/yolov8-object-detection/runs/detect",
|
| 187 |
+
name="train_yolov8s",
|
| 188 |
+
exist_ok=True,
|
| 189 |
+
patience=200,
|
| 190 |
+
lr0=0.01,
|
| 191 |
+
momentum=0.937,
|
| 192 |
+
weight_decay=0.0005
|
| 193 |
+
)
|
| 194 |
+
```
|
| 195 |
+
|
| 196 |
+
---
|
| 197 |
+
|
| 198 |
+
### ✅ **Validación del Modelo**
|
| 199 |
+
Después del entrenamiento, validamos el modelo con el siguiente código:
|
| 200 |
+
|
| 201 |
+
```python
|
| 202 |
+
from ultralytics import YOLO
|
| 203 |
+
|
| 204 |
+
# Cargar el modelo entrenado
|
| 205 |
+
model = YOLO("/home/USER/yolo/yolov8-object-detection/runs/detect/train_yolov8s/weights/best.pt")
|
| 206 |
+
|
| 207 |
+
# Validar el modelo y guardar los resultados
|
| 208 |
+
metrics = model.val(
|
| 209 |
+
data="/home/USER/yolo/yolov8-object-detection/datasets/combined/data.yaml",
|
| 210 |
+
project="/home/USER/yolo/yolov8-object-detection/runs/val",
|
| 211 |
+
name="val",
|
| 212 |
+
exist_ok=True
|
| 213 |
+
)
|
| 214 |
+
|
| 215 |
+
print(metrics)
|
| 216 |
+
```
|
| 217 |
+
|
| 218 |
+
### 🔹 **Recomendación**
|
| 219 |
+
Primeramente descarga el dataset de **COCO8** para que genere la estructura correcta de los archivos para incluir los demás datasets.
|
| 220 |
+
|
| 221 |
+
*Para juntar varios datasets habrá que hacerlo manualmente o mediante un codigo de python que añada las imágenes y labels a sus carpetas correspondientes.*
|
| 222 |
+
|
| 223 |
+
---
|
| 224 |
+
|
| 225 |
+
### 💻 **Uso del Modelo en Videos**
|
| 226 |
+
Para aplicar el modelo a un video y detectar objetos:
|
| 227 |
+
|
| 228 |
+
```python
|
| 229 |
+
from ultralytics import YOLO
|
| 230 |
+
|
| 231 |
+
# Cargar el modelo entrenado
|
| 232 |
+
model = YOLO("runs/detect/train_yolov8s/weights/best.pt")
|
| 233 |
+
|
| 234 |
+
# Realizar inferencia en un video
|
| 235 |
+
results = model.predict("ruta/video.mp4", save=True, conf=0.5)
|
| 236 |
+
|
| 237 |
+
# Guardar el video con las detecciones
|
| 238 |
+
print("✅ Procesamiento completado. Video guardado.")
|
| 239 |
+
```
|
| 240 |
+
|
| 241 |
+
---
|
| 242 |
+
|
| 243 |
+
### 📂 Estructura del Proyecto YOLOv8
|
| 244 |
+
|
| 245 |
+
```bash
|
| 246 |
+
.
|
| 247 |
+
├── .venv/ # Entorno
|
| 248 |
+
├── datasets/ # Carpeta de datasets
|
| 249 |
+
│ ├── coco8/ # Dataset COCO8
|
| 250 |
+
│ ├── yolov8s.pt # Pesos preentrenados de YOLOv8s
|
| 251 |
+
├── datasets-download/ # Descargas de datasets
|
| 252 |
+
├── processed-video/ # Vídeos procesados
|
| 253 |
+
│ ├── ny-traffic-processed.mp4 # Vídeo de tráfico procesado
|
| 254 |
+
├── raw-video/ # Vídeos sin procesar
|
| 255 |
+
│ ├── ny-traffic.mp4 # Vídeo de tráfico original
|
| 256 |
+
├── runs/ # Resultados de entrenamiento y validación
|
| 257 |
+
│ ├── detect/ # Carpeta de detección de objetos
|
| 258 |
+
│ │ ├── train_coco8 # Entrenamiento con COCO8
|
| 259 |
+
│ │ ├── train_yolov8n # Entrenamiento con YOLOv8n
|
| 260 |
+
│ ├── val/ # Resultados de validación
|
| 261 |
+
│ │ ├─��� val_coco8/ # Resultados de validación con gráficas relevantes
|
| 262 |
+
│ │ ├── best.pt # Mejor peso del modelo YOLOv8s entrenado con COCO8
|
| 263 |
+
│ │ ├── last.pt # Último peso del modelo YOLOv8s entrenado con COCO8
|
| 264 |
+
├── .gitattributes # Configuración de atributos de Git
|
| 265 |
+
├── .gitignore # Ignorar archivos innecesarios en Git
|
| 266 |
+
├── predict.py # Script para realizar predicciones a los vídeos
|
| 267 |
+
├── requirements.txt # Dependencias del proyecto
|
| 268 |
+
├── train_yolov8n.py # Script para entrenar YOLOv8n
|
| 269 |
+
├── train_yolov8s.py # Script para entrenar YOLOv8s
|
| 270 |
+
├── validate.py # Script para validar el modelo
|
| 271 |
+
├── yolov8n.pt # Pesos del modelo YOLOv8n
|
| 272 |
+
```
|
| 273 |
+
|
| 274 |
+
---
|
| 275 |
+
|
| 276 |
+
### 📊 **Resultados y Gráficos**
|
| 277 |
+
|
| 278 |
+
#### Comparación General de Resultados entre YOLOv8n y YOLOv8s
|
| 279 |
+
|
| 280 |
+
| **Métrica** | **YOLOv8n** | **YOLOv8s** | **Diferencia (YOLOv8s - YOLOv8n)** |
|
| 281 |
+
|--------------------|------------|------------|----------------------------------|
|
| 282 |
+
| **Precisión (B)** | 0.748 | 0.821 | +0.073 |
|
| 283 |
+
| **Recall (B)** | 0.561 | 0.920 | +0.359 |
|
| 284 |
+
| **mAP@50 (B)** | 0.645 | 0.944 | +0.299 |
|
| 285 |
+
| **mAP@50-95 (B)** | 0.431 | 0.726 | +0.295 |
|
| 286 |
+
| **Fitness** | 0.453 | 0.747 | +0.294 |
|
| 287 |
+
|
| 288 |
+
#### 📌 Análisis:
|
| 289 |
+
- **Precisión**: YOLOv8s tiene mejor precisión (**+7.3%**), lo que significa que el modelo comete menos falsos positivos.
|
| 290 |
+
- **Recall**: YOLOv8s tiene un recall significativamente mayor (**+35.9%**), indicando que detecta más objetos correctamente.
|
| 291 |
+
- **mAP@50**: YOLOv8s supera a YOLOv8n en un **30%**, lo que sugiere que el modelo más grande tiene una mejor capacidad para detectar objetos con alta confianza.
|
| 292 |
+
- **mAP@50-95**: También mejora en un **29.5%**, lo que significa que tiene un rendimiento más consistente en diferentes umbrales de IoU.
|
| 293 |
+
- **Fitness**: YOLOv8s tiene una mejora notable (**+29.4%**), lo que indica un mejor balance entre precisión y recall.
|
| 294 |
+
|
| 295 |
+
---
|
| 296 |
+
|
| 297 |
+
#### Comparación de Velocidad
|
| 298 |
+
|
| 299 |
+
| **Parámetro** | **YOLOv8n** | **YOLOv8s** | **Diferencia** |
|
| 300 |
+
|---------------------|------------|------------|----------------|
|
| 301 |
+
| **Preprocesamiento** | 1.92 ms | 1.68 ms | -0.24 ms |
|
| 302 |
+
| **Inferencia** | 55.05 ms | 128.99 ms | +73.94 ms |
|
| 303 |
+
| **Postprocesamiento** | 1.22 ms | 0.91 ms | -0.31 ms |
|
| 304 |
+
|
| 305 |
+
#### 📌 Análisis:
|
| 306 |
+
- **Preprocesamiento**: Similar en ambos modelos.
|
| 307 |
+
- **Inferencia**: **YOLOv8s es mucho más lento** en inferencia (**+74 ms**), lo cual es esperable ya que es un modelo más grande.
|
| 308 |
+
- **Postprocesamiento**: Ligeramente más rápido en **YOLOv8s**, pero la diferencia no es significativa.
|
| 309 |
+
|
| 310 |
+
---
|
| 311 |
+
### 🔍 YOLOv8n vs YOLOv8s
|
| 312 |
+
|
| 313 |
+
#### Comparación de Métricas
|
| 314 |
+
|
| 315 |
+

|
| 316 |
+
|
| 317 |
+
#### 📌 Análisis:
|
| 318 |
+
- YOLOv8s supera en todas las métricas a YOLOv8n.
|
| 319 |
+
- La mayor diferencia se observa en **Recall (+35.9%)** y **mAP@50-95 (+29.5%)**, indicando una mejor detección a diferentes umbrales de IoU.
|
| 320 |
+
- Aunque YOLOv8s tiene mejor rendimiento, su velocidad de inferencia es más lenta.
|
| 321 |
+
|
| 322 |
+
---
|
| 323 |
+
|
| 324 |
+
#### Evolución de la Función de Pérdida por Época
|
| 325 |
+
|
| 326 |
+

|
| 327 |
+
|
| 328 |
+
#### 📌 Análisis:
|
| 329 |
+
- Ambos modelos muestran una disminución de la pérdida a lo largo del entrenamiento.
|
| 330 |
+
- **YOLOv8s** comienza con una pérdida mayor pero converge bien, sugiriendo que aprende mejor con más iteraciones.
|
| 331 |
+
- **YOLOv8n** tiene una convergencia más rápida pero con menor precisión general.
|
| 332 |
+
|
| 333 |
+
---
|
| 334 |
+
|
| 335 |
+
#### Curva de Precisión-Recall
|
| 336 |
+
|
| 337 |
+

|
| 338 |
+
|
| 339 |
+
#### 📌 Análisis:
|
| 340 |
+
- **YOLOv8s** mantiene una precisión más alta en todos los valores de recall, lo que significa menos falsos positivos en comparación con YOLOv8n.
|
| 341 |
+
- **YOLOv8n** muestra más fluctuaciones en la curva, indicando menor estabilidad en la detección de objetos.
|
| 342 |
+
|
| 343 |
+
---
|
| 344 |
+
|
| 345 |
+
#### Matriz de Confusión
|
| 346 |
+
|
| 347 |
+

|
| 348 |
+
|
| 349 |
+
#### 📌 Análisis:
|
| 350 |
+
- Ambas matrices muestran que algunos objetos están siendo confundidos entre sí.
|
| 351 |
+
- **YOLOv8s** presenta menos errores de clasificación en comparación con YOLOv8n.
|
| 352 |
+
- La normalización de la matriz confirma que YOLOv8s tiene una mejor distribución de predicciones.
|
| 353 |
+
|
| 354 |
+
---
|
| 355 |
+
|
| 356 |
+
#### Ejemplo de Detección
|
| 357 |
+
|
| 358 |
+

|
| 359 |
+
|
| 360 |
+
---
|
| 361 |
+
>>>>>>> yolov8-model
|
predict.py
ADDED
|
@@ -0,0 +1,36 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import cv2
|
| 2 |
+
from ultralytics import YOLO
|
| 3 |
+
|
| 4 |
+
# Cargar modelo YOLOv8 entrenado
|
| 5 |
+
model = YOLO("/home/izaskunmz/yolo/yolov8-object-detection/runs/detect/train_coco8/weights/best.pt")
|
| 6 |
+
|
| 7 |
+
# Abrir vídeo
|
| 8 |
+
video_path = "/home/izaskunmz/yolo/yolov8-object-detection/raw-video/ny-traffic.mp4"
|
| 9 |
+
cap = cv2.VideoCapture(video_path)
|
| 10 |
+
|
| 11 |
+
# Obtener dimensiones del video original
|
| 12 |
+
width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
|
| 13 |
+
height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
|
| 14 |
+
fps = int(cap.get(cv2.CAP_PROP_FPS))
|
| 15 |
+
|
| 16 |
+
# Definir el codec y crear el VideoWriter para guardar el resultado
|
| 17 |
+
output_path = "/home/izaskunmz/yolo/yolov8-object-detection/processed-video/ny-traffic-processed.mp4"
|
| 18 |
+
fourcc = cv2.VideoWriter_fourcc(*'mp4v') # Codec para formato MP4
|
| 19 |
+
out = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
|
| 20 |
+
|
| 21 |
+
while cap.isOpened():
|
| 22 |
+
ret, frame = cap.read()
|
| 23 |
+
if not ret:
|
| 24 |
+
break # Si el vídeo ha terminado, salimos del bucle
|
| 25 |
+
|
| 26 |
+
# Realizar detección en el frame
|
| 27 |
+
results = model(frame)
|
| 28 |
+
|
| 29 |
+
# Obtener frame con anotaciones
|
| 30 |
+
annotated_frame = results[0].plot()
|
| 31 |
+
|
| 32 |
+
# Guardar el frame en el video de salida
|
| 33 |
+
out.write(annotated_frame)
|
| 34 |
+
|
| 35 |
+
cap.release()
|
| 36 |
+
out.release() # Liberar el escritor de video
|
processed-video/ny-traffic-processed.mp4
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ae66dfe798c8d870971336d36f5a2e2889416966ac8e1061760de1c26e22fd03
|
| 3 |
+
size 158863855
|
processed-video/ny-walking-processed.mp4
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:37dc741a42b183c4feadd7108fea2db6dc6b6116f9ac596f92944c7418e9cb4d
|
| 3 |
+
size 178527503
|
raw-video/ny-traffic.mp4
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:03bb512b109a238d9bdacfb118ce45e312925a1283f06c647624408573f4112a
|
| 3 |
+
size 62404880
|
raw-video/ny-walking.mp4
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:992ce64943371f674fdc2b243fc1228985e8d387e387230428e8c57a071d8e8c
|
| 3 |
+
size 20162842
|
requirements.txt
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
ultralytics
|
runs/val/best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:dfa11240a586e524e3b42e818f22b8afa899fa9b2425ca69648783f421d43829
|
| 3 |
+
size 22592803
|
runs/val/last.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1314b2c5f9e3c55ff97fdbc023b0531b7d4217a00420698c731ec4e9278c6e35
|
| 3 |
+
size 22592803
|
runs/val/val_coco8/F1_curve.png
ADDED
|
runs/val/val_coco8/PR_curve.png
ADDED
|
runs/val/val_coco8/P_curve.png
ADDED
|
runs/val/val_coco8/R_curve.png
ADDED
|
runs/val/val_coco8/confusion_matrix.png
ADDED
|
runs/val/val_coco8/confusion_matrix_normalized.png
ADDED
|
runs/val/val_coco8/val_batch0_labels.jpg
ADDED
|
runs/val/val_coco8/val_batch0_pred.jpg
ADDED
|
train_yolov8n.py
ADDED
|
@@ -0,0 +1,24 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from ultralytics import YOLO
|
| 2 |
+
|
| 3 |
+
# Cargar el modelo YOLOv8 preentrenado
|
| 4 |
+
model = YOLO("yolov8n.pt") # Puedes probar con "yolov8s.pt" para mayor precisión
|
| 5 |
+
|
| 6 |
+
# Entrenar el modelo con hiperparámetros ajustados
|
| 7 |
+
model.train(
|
| 8 |
+
data="/home/izaskunmz/yolo/yolov8-object-detection/datasets/coco8/data.yaml", # Ruta correcta al dataset
|
| 9 |
+
epochs=150, # Aumentamos las épocas para mejorar precisión
|
| 10 |
+
batch=8, # Reducimos el batch para estabilidad en CPU
|
| 11 |
+
imgsz=640, # Tamaño de la imagen
|
| 12 |
+
device="cpu", # Entrenamiento en CPU
|
| 13 |
+
lr0=0.0005, # Learning rate inicial más bajo para mejorar estabilidad
|
| 14 |
+
lrf=0.0001, # Decaimiento más lento del learning rate
|
| 15 |
+
momentum=0.95, # Aumentamos momentum para estabilizar entrenamiento
|
| 16 |
+
weight_decay=0.0001, # Regularización más fuerte para evitar sobreajuste
|
| 17 |
+
optimizer="AdamW", # Mejor optimizador que SGD para convergencia en CPU
|
| 18 |
+
cos_lr=True, # Usamos learning rate decay con coseno para ajuste fino
|
| 19 |
+
close_mosaic=5, # Desactivamos aumentación mosaico después de 5 épocas
|
| 20 |
+
patience=0, # 🔹 Desactiva Early Stopping
|
| 21 |
+
project="/home/izaskunmz/yolo/yolov8-object-detection/runs/detect", # Ruta correcta para guardar los modelos
|
| 22 |
+
name="train_yolov8n", # Nombre del experimento optimizado
|
| 23 |
+
exist_ok=True # Evita sobreescritura, crea versiones numeradas
|
| 24 |
+
)
|
train_yolov8s.py
ADDED
|
@@ -0,0 +1,20 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from ultralytics import YOLO
|
| 2 |
+
|
| 3 |
+
# Cargar el modelo YOLOv8s preentrenado
|
| 4 |
+
model = YOLO("yolov8s.pt")
|
| 5 |
+
|
| 6 |
+
# Entrenar el modelo y guardar en la carpeta correcta
|
| 7 |
+
model.train(
|
| 8 |
+
data="/home/izaskunmz/yolo/yolov8-object-detection/datasets/coco8/data.yaml", # Archivo de configuración del dataset
|
| 9 |
+
epochs=150, # Aumentamos las épocas para mejorar el aprendizaje
|
| 10 |
+
batch=8, # Reducimos el batch si hay problemas de memoria
|
| 11 |
+
imgsz=640, # Tamaño de las imágenes
|
| 12 |
+
device='cpu', # Si tienes GPU, cámbialo a 'cuda'
|
| 13 |
+
project="/home/izaskunmz/yolo/yolov8-object-detection/runs/detect", # Carpeta donde se guardarán los resultados
|
| 14 |
+
name="train_coco8", # Nombre del experimento
|
| 15 |
+
exist_ok=True, # Si la carpeta existe, crea una nueva numerada
|
| 16 |
+
patience=200, # Para evitar que se detenga temprano
|
| 17 |
+
lr0=0.01, # Ajustamos la tasa de aprendizaje inicial
|
| 18 |
+
momentum=0.937, # Momentum del optimizador
|
| 19 |
+
weight_decay=0.0005 # Regularización para evitar overfitting
|
| 20 |
+
)
|
validate.py
ADDED
|
@@ -0,0 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from ultralytics import YOLO
|
| 2 |
+
|
| 3 |
+
# Cargar el modelo entrenado
|
| 4 |
+
model = YOLO("/home/izaskunmz/yolo/yolov8-object-detection/runs/detect/train_coco8/weights/best.pt") # Asegúrate de que esta ruta sea correcta
|
| 5 |
+
|
| 6 |
+
# Validar el modelo y guardar los resultados en la carpeta correcta
|
| 7 |
+
metrics = model.val(
|
| 8 |
+
data="/home/izaskunmz/yolo/yolov8-object-detection/datasets/coco8/data.yaml",
|
| 9 |
+
project="/home/izaskunmz/yolo/yolov8-object-detection/runs/val", # Define la carpeta donde se guardarán los resultados
|
| 10 |
+
name="val_coco8", # Nombre del experimento
|
| 11 |
+
exist_ok=True # Evita sobreescribir, creará nuevas versiones numeradas
|
| 12 |
+
)
|
| 13 |
+
|
| 14 |
+
# Mostrar las métricas de evaluación
|
| 15 |
+
print(metrics)
|