chore: 更新最新代码

This commit is contained in:
琉璃月光
2025-08-13 14:49:06 +08:00
parent 4716e5f2e2
commit da205bf47c
1123 changed files with 990 additions and 3 deletions

View File

@ -0,0 +1,22 @@
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, 2], # [数量, 坐标维度],您的任务是 4 个关键点 (x,y)
)