22 lines
633 B
Python
22 lines
633 B
Python
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)
|
||
) |