20 lines
503 B
Python
20 lines
503 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=300,
|
|||
|
|
batch=4,
|
|||
|
|
workers=10,
|
|||
|
|
device='0',
|
|||
|
|
project='runs/train/point',
|
|||
|
|
name='exp_pose',
|
|||
|
|
exist_ok=False,
|
|||
|
|
optimizer='AdamW',
|
|||
|
|
lr0=0.001,
|
|||
|
|
patience=100,
|
|||
|
|
)
|