# train_merged.py
# -*- coding: utf-8 -*-
"""
合并数据集训练脚本
在服务器上运行：python train_merged.py
"""

from ultralytics import YOLO

# 使用标准 YOLO11l-seg 模型（不要用 YOLOE！）
model = YOLO("/home/rongye/ProgramFiles/Yolo/yolo11l-seg.pt")

# 开始训练
results = model.train(
    data="/home/rongye/ProgramFiles/Yolo/datasets/blind_guidance_merged/data.yaml",
    epochs=200,           # 数据量大，增加轮次
    imgsz=640,            # 训练分辨率
    batch=16,             # 3090 显存足够
    device=1,             # 第二块 3090
    workers=16,           # 多线程加载
    cache='ram',          # 192GB 内存全量缓存
    optimizer='AdamW',
    close_mosaic=15,      # 最后15轮关闭mosaic
    amp=False,            # 混合精度 (NaN时关闭)
    patience=30,          # 早停耐心值
    project="blind_guide_project",
    name="yolo11l_blind_v2"  # 版本2
)