chore: 更新最新代码
This commit is contained in:
22
ultralytics_yolo11-main/train_point_main.py
Normal file
22
ultralytics_yolo11-main/train_point_main.py
Normal 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)
|
||||
)
|
||||
Reference in New Issue
Block a user