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 |
)
|