from ultralytics import YOLO

if __name__ == '__main__':
    # ✅ 使用官方支持的关键点模型（YOLOv11）
    model = YOLO(r'ultralytics/cfg/models/11/yolo11-pose.yaml')  # 或

    results = model.train(
        data='point_data.yaml',
        imgsz=1280,
        epochs=300,
        batch=4,
        workers=10,
        device='0',
        project='runs/train/point',
        name='exp_pose',
        exist_ok=False,
        optimizer='AdamW',
        lr0=0.001,
        patience=100,
    )