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=100,
        batch=4,
        workers=10,
        device='0',
        project='runs/train',
        name='exp_pose',
        exist_ok=False,
        optimizer='AdamW',
        lr0=0.001,
        patience=20,
        # ✅ 关键：指定关键点数量
        #kpt_shape=[4, 3],  # [数量, 坐标维度]，您的任务是 4 个关键点 (x,y)
    )