xixu-me/yolo-tinyperson
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update train.py

Browse Source
This commit is contained in:
xixu-me committed 2025-05-27 09:50:04 +08:00
1 parent cd8318fba5
commit 08bd87a9db
1 file changed
+5 -5
Download Patch File Download Diff File Expand all files Collapse all files
train.py
+5 -5
View File
@@ -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},
]