izaskunmz commited on
Commit
0875865
·
1 Parent(s): 2cf8ba5

delete readme and modified train_yolo8n

Browse files
Files changed (1) hide show
  1. 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=300, # 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,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="train_optimized", # Nombre del experimento optimizado
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
  )