izaskunmz
commited on
Commit
·
0875865
1
Parent(s):
2cf8ba5
delete readme and modified train_yolo8n
Browse files- train_yolov8n.py +2 -2
train_yolov8n.py
CHANGED
|
@@ -6,7 +6,7 @@ model = YOLO("yolov8n.pt") # Puedes probar con "yolov8s.pt" para mayor precisi
|
|
| 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=
|
| 10 |
batch=8, # Reducimos el batch para estabilidad en CPU
|
| 11 |
imgsz=640, # Tamaño de la imagen
|
| 12 |
device="cpu", # Entrenamiento en CPU
|
|
@@ -19,6 +19,6 @@ model.train(
|
|
| 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="
|
| 23 |
exist_ok=True # Evita sobreescritura, crea versiones numeradas
|
| 24 |
)
|
|
|
|
| 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
|
|
|
|
| 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 |
)
|