From 08bd87a9db040043ba5cb22f52c7ee5c9cedc072 Mon Sep 17 00:00:00 2001 From: Xi Xu Date: Tue, 27 May 2025 09:50:04 +0800 Subject: [PATCH] Update train.py --- train.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/train.py b/train.py index f94fbfd..6282c8a 100644 --- a/train.py +++ b/train.py @@ -5,11 +5,11 @@ from ultralytics import YOLO # ===== Model Configuration ===== MODELS = [ - {"name": "YOLOv8x", "model_path": "weights/yolov8x.pt", "epochs": 2}, - {"name": "YOLOv9e", "model_path": "weights/yolov9e.pt", "epochs": 2}, - {"name": "YOLOv10x", "model_path": "weights/yolov10x.pt", "epochs": 2}, - {"name": "YOLO11x", "model_path": "weights/yolo11x.pt", "epochs": 2}, - {"name": "YOLO12x", "model_path": "weights/yolo12x.pt", "epochs": 2}, + {"name": "YOLOv8x", "model_path": "weights/yolov8x.pt", "epochs": 100}, + {"name": "YOLOv9e", "model_path": "weights/yolov9e.pt", "epochs": 100}, + {"name": "YOLOv10x", "model_path": "weights/yolov10x.pt", "epochs": 100}, + {"name": "YOLO11x", "model_path": "weights/yolo11x.pt", "epochs": 100}, + {"name": "YOLO12x", "model_path": "weights/yolo12x.pt", "epochs": 100}, ]